Hi , xintao. 1. I have seen that the parameter of pixel_weight is 0 in train_stfgan.json not 1e-2 as in in train_esrgan.json or train_srgan.json. Is this right ? 2. I think the parameter of use_rot is true not false as in your original train_stfgan.json file. Best regards .