How to train x2 model?

Open yangshunDragon opened this issue 5 years ago • 2 comments

Hello, I use kernelGAN to generate x4 kernels. If I want to train x2 model, should I re-generate x2 kernels with kernelGAN with --X2 parameters? or could I prepare the datapair by x4 kernels? What's more, could I finetune x2 model from pretrained RRDB_PSNR_x4.pth model ?

Jan 07 '21 05:01 yangshunDragon

Hello, I use kernelGAN to generate x4 kernels. If I want to train x2 model, should I re-generate x2 kernels with kernelGAN with --X2 parameters? or could I prepare the datapair by x4 kernels? What's more, could I finetune x2 model from pretrained RRDB_PSNR_x4.pth model ?

hello, do you know how to train x2 model now ??

Jan 19 '21 03:01 fourPieces0927

According to the source code, remove "--X4" in "CUDA_VISIBLE_DEVICES=4,5,6,7 python3 train.py --X4 --input-dir SOURCE_PATH" will meet your expectation

Mar 02 '21 06:03 syfbme