From 84d4d31dc6fded754378a51615b25a6222325ca3 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Sat, 2 Mar 2024 19:36:48 +0000 Subject: [PATCH] fix fastmri datasetup --- datasets/dataset_setup.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/datasets/dataset_setup.py b/datasets/dataset_setup.py index b22d352ce..1d3e5d2c7 100644 --- a/datasets/dataset_setup.py +++ b/datasets/dataset_setup.py @@ -706,13 +706,13 @@ def main(_): 'to download the FastMRI dataset.\nSign up for the URLs at ' 'https://fastmri.med.nyu.edu/.') - updated_data_dir = download_fastmri(data_dir, - knee_singlecoil_train_url, - knee_singlecoil_val_url, - knee_singlecoil_test_url) + download_fastmri(data_dir, + knee_singlecoil_train_url, + knee_singlecoil_val_url, + knee_singlecoil_test_url) logging.info('fastMRI download completed. Extracting...') - setup_fastmri(data_dir, updated_data_dir) + setup_fastmri(data_dir) if FLAGS.all or FLAGS.imagenet: flags.mark_flag_as_required('imagenet_train_url')