From 04c18832ab6611eae9a577235543cfc357014a77 Mon Sep 17 00:00:00 2001
From: sakares <s.sakares@gmail.com>
Date: Wed, 20 Nov 2024 17:36:28 +0800
Subject: [PATCH] add to_buffer at the end of data_iter

---
 speechcommands/main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/speechcommands/main.py b/speechcommands/main.py
index 0d8da9fd..3f0de98c 100644
--- 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