diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py index 60710ceb72e5828182018e3cfc3097e6bca5b1ad..94414237dda5cf0334c59492720b0fc403d3c376 100644 --- a/modules/GSOC/E2_ESRGAN/lib/train.py +++ b/modules/GSOC/E2_ESRGAN/lib/train.py @@ -40,7 +40,8 @@ class Trainer(object): hr_size = dataset_args['hr_dimension'] lr_size = int(hr_size // 4) - filenames = glob.glob('/Users/tomrink/data/opd_singles/valid_ires_??1?.npy') + # filenames = glob.glob('/Users/tomrink/data/opd_singles/valid_ires_??1?.npy') + filenames = glob.glob('/ships22/cloud/scratch/Satellite_Output/GOES-16/global/NREL_2023/2023_east_cf/clavrx_surfrad_abi_cld_opd_1_512x512/valid_ires_????.npy') self.dataset = dataset.OpdNpyDataset(filenames, hr_size, lr_size, batch_size=self.batch_size).dataset print('OPD dataset initialized...')