Explicity set the shuffle_buffer_size parameter to address meta-datas…

…et issue #54 (google-research/meta-dataset#54).
cambridge-mlg · Dec 29, 2020 · 91a0414 · 91a0414
1 parent 7671893
commit 91a0414
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/learners/cnaps/src/meta_dataset_reader.py b/learners/cnaps/src/meta_dataset_reader.py
@@ -85,7 +85,8 @@ def _init_single_source_dataset(self, dataset_name, split, episode_description):
             use_bilevel_ontology=use_bilevel_ontology,
             split=split,
             episode_descr_config=episode_description,
-            image_size=84)
+            image_size=84,
+            shuffle_buffer_size=1000)
 
         iterator = single_source_pipeline.make_one_shot_iterator()
         return iterator.get_next()
@@ -174,7 +175,8 @@ def _init_dataset(self, dataset, split, episode_description):
             use_bilevel_ontology=False,
             split=split,
             episode_descr_config=episode_description,
-            image_size=84)
+            image_size=84,
+            shuffle_buffer_size=1000)
 
         iterator = single_source_pipeline.make_one_shot_iterator()
         return iterator.get_next()