diff --git a/modules/GSOC/E2_ESRGAN/config/config.yaml b/modules/GSOC/E2_ESRGAN/config/config.yaml index c70e810f4365e771a31362053efbd556fc1d6bf7..7f55d4325965222f988d2438ba8361cf085cea75 100644 --- a/modules/GSOC/E2_ESRGAN/config/config.yaml +++ b/modules/GSOC/E2_ESRGAN/config/config.yaml @@ -16,7 +16,7 @@ RDB: residual_scale_beta: 0.2 #Training # warmup_num_iter: 1000 # Number of steps to train the PSNR model -batch_size: 4 +batch_size: 8 train_psnr: num_steps: 600000 adam: