fix

Author: zsdonghao

config.py

@@ -5,7 +5,7 @@
 config.TRAIN = edict()
 
 ## Adam
-config.TRAIN.batch_size = 16
+config.TRAIN.batch_size = 8 # [16] use 8 if your GPU memory is small, and use [2, 4] in tl.vis.save_images / use 16 for faster training
 config.TRAIN.lr_init = 1e-4
 config.TRAIN.beta1 = 0.9
 
@@ -20,13 +20,13 @@
 config.TRAIN.decay_every = int(config.TRAIN.n_epoch / 2)
 
 ## train set location
-config.TRAIN.hr_img_path = 'data2017/DIV2K_train_HR/'
-config.TRAIN.lr_img_path = 'data2017/DIV2K_train_LR_bicubic/X4/'
+config.TRAIN.hr_img_path = 'DIV2K/DIV2K_train_HR/'
+config.TRAIN.lr_img_path = 'DIV2K/DIV2K_train_LR_bicubic/X4/'
 
 config.VALID = edict()
 ## test set location
-config.VALID.hr_img_path = 'data2017/DIV2K_valid_HR/'
-config.VALID.lr_img_path = 'data2017/DIV2K_valid_LR_bicubic/X4/'
+config.VALID.hr_img_path = 'DIV2K/DIV2K_valid_HR/'
+config.VALID.lr_img_path = 'DIV2K/DIV2K_valid_LR_bicubic/X4/'
 
 def log_config(filename, cfg):
     with open(filename, 'w') as f: