diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py index 64e7e98a9a2119844d9c073f396869fed6b8120c..d05c56939da4cdd53ecd769f862f4e37f02e8a45 100644 --- a/modules/GSOC/E2_ESRGAN/lib/dataset.py +++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py @@ -160,7 +160,7 @@ class OpdNpyDataset: dataset = dataset.shuffle(2000, reshuffle_each_iteration=True) dataset = dataset.batch(batch_size) dataset = dataset.map(self.data_function, num_parallel_calls=8) - dataset = dataset.cache(filename='/ships22/cloud/scratch/Satellite_Output/GOES-16/global/NREL_2023/2023_east_cf/cld_opd_train') + dataset = dataset.cache(filename='/ships22/cloud/scratch/Satellite_Output/GOES-16/global/NREL_2023/2023_east_cf/cld_opd_train/cld_opd_train') dataset = dataset.prefetch(buffer_size=1) self.dataset = dataset