add to_buffer at the end of data_iter

2025-08-29 18:26:37 +08:00 · 2024-11-20 17:36:28 +08:00 · 2024-11-20 17:36:28 +08:00 · 04c18832ab
commit 04c18832ab
parent bd6d910ca3
1 changed files with 1 additions and 0 deletions
--- a/speechcommands/main.py
+++ b/speechcommands/main.py
@ -48,6 +48,7 @@ def prepare_dataset(batch_size, split, root=None):
        .batch(batch_size)
        .to_stream()
        .prefetch(4, 4)
+        .to_buffer()
    )
    return data_iter