diff --git a/modules/GSOC/E2_ESRGAN/lib/dataset.py b/modules/GSOC/E2_ESRGAN/lib/dataset.py index ce9f4495c8d4b90cf331c2e7e6a01182cd1c3384..be07dd414f0ad2bfb20443ed935e35b308226c16 100644 --- a/modules/GSOC/E2_ESRGAN/lib/dataset.py +++ b/modules/GSOC/E2_ESRGAN/lib/dataset.py @@ -133,6 +133,8 @@ f = open(mean_std_file, 'rb') mean_std_dct_l1b = pickle.load(f) f.close() +mean_std_dct.update(mean_std_dct_l1b) +mean_std_dct.update(mean_std_dct_l2) class OpdNpyDataset: