diff --git a/modules/GSOC/E2_ESRGAN/config/config.yaml b/modules/GSOC/E2_ESRGAN/config/config.yaml
index ab4afa5fda4c29bf064dd9425db58c5ed40206c8..d85944b517c1f4eb3536a39af2991508870cbba2 100644
--- a/modules/GSOC/E2_ESRGAN/config/config.yaml
+++ b/modules/GSOC/E2_ESRGAN/config/config.yaml
@@ -3,7 +3,7 @@ checkpoint_path:
   phase_2: "checkpoints/phase_2"
 
 dataset: 
-  name: "div2k"
+  name: "cld_opd_dcomp"
   scale_method: "bicubic"
   hr_dimension: 512