diff --git a/modules/GSOC/E2_ESRGAN/lib/train.py b/modules/GSOC/E2_ESRGAN/lib/train.py
index a6775e1c73bc95b62f84c1e8909d77ba82458f30..1372e3802073bedf074299b56e756707640c2dfc 100644
--- a/modules/GSOC/E2_ESRGAN/lib/train.py
+++ b/modules/GSOC/E2_ESRGAN/lib/train.py
@@ -13,6 +13,7 @@ logging.basicConfig(filename=str(Path.home()) + '/esrgan_log.txt', level=logging
 
 NUM_WU_EPOCHS = 60
 NUM_EPOCHS = 80
+num_chans = 2
 
 
 class Trainer(object):