zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 07:30:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add to_buffer at the end of data_iter

Browse Source
This commit is contained in:
sakares 2024-11-20 17:36:28 +08:00
parent bd6d910ca3
commit 04c18832ab
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
speechcommands/main.py
View File

@ -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