revert fastmri input pipeline change

mlcommons · Oct 23, 2023 · 730be20 · 730be20
1 parent c096464
commit 730be20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/algorithmic_efficiency/workloads/fastmri/input_pipeline.py b/algorithmic_efficiency/workloads/fastmri/input_pipeline.py
@@ -180,7 +180,7 @@ def load_fastmri_split(global_batch_size,
     h5_paths = glob.glob(file_pattern)
   elif split == 'validation':
     file_pattern = os.path.join(data_dir, _VAL_DIR, '*.h5')
-    h5_paths = sorted(glob.glob(file_pattern))
+    h5_paths = sorted(glob.glob(file_pattern))[:100]
   elif split == 'test':
     # The fastmri validation set is split into a validation and test set
     file_pattern = os.path.join(data_dir, _VAL_DIR, '*.h5')