diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py
index a98228468d154cd9c9e8b71e9f30e3e7ec787beb..64e7e98a9a2119844d9c073f396869fed6b8120c 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()
+        dataset = dataset.cache(filename='/ships22/cloud/scratch/Satellite_Output/GOES-16/global/NREL_2023/2023_east_cf/cld_opd_train')
         dataset = dataset.prefetch(buffer_size=1)
 
         self.dataset = dataset