Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to train x2 model? #15

Open
yangshunDragon opened this issue Jan 7, 2021 · 2 comments
Open

How to train x2 model? #15

yangshunDragon opened this issue Jan 7, 2021 · 2 comments

Comments

@yangshunDragon
Copy link

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 ?

@fourPieces0927
Copy link

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 ??

@syfbme
Copy link

syfbme commented Mar 2, 2021

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants