From 730be20c1317b06c682f3709fd09e0336b351326 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Mon, 23 Oct 2023 18:26:28 +0000 Subject: [PATCH] revert fastmri input pipeline change --- algorithmic_efficiency/workloads/fastmri/input_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/algorithmic_efficiency/workloads/fastmri/input_pipeline.py b/algorithmic_efficiency/workloads/fastmri/input_pipeline.py index df59098e4..8f6ddafd1 100644 --- 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')