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

a bug that during training the x2 model, 0811.pt is not generated? #4

Open
xiaoxiongli opened this issue Jun 1, 2021 · 6 comments
Open

Comments

@xiaoxiongli
Copy link

xiaoxiongli commented Jun 1, 2021

Hi Dear @NJU-Jet :
it seems have a bug, please see below figure:
image

image

and I check the folder data/DIV2K/bin/DIV2K_train_HR/, below figure:

image

you can see that the it have 0001.pt ~ 0810.pt, and no 0811.pt ... , so the train.py crashed...

How to fix this? thank you~

@NJU-Jet
Copy link
Owner

NJU-Jet commented Jun 1, 2021

If you want to only use ten images for fast validation, you can modify DIV2K_val.txt file. Retain the first ten lines and remove other lines.

@xiaoxiongli
Copy link
Author

OK, I got it, thank you and below warning is normal or not?
image

@NJU-Jet
Copy link
Owner

NJU-Jet commented Jun 1, 2021

It's normal. Sometimes PSNR doesn't increase and you should retrain the model.

@xiaoxiongli
Copy link
Author

And For 0811.pt is not generated, I find the reason is the README.md's data tree is not same as the base7.yaml, the base7.yaml have another "bin" while README.md not have. I change the base7.yaml, it works.

@xiaoxiongli
Copy link
Author

xiaoxiongli commented Jun 1, 2021

It's normal. Sometimes PSNR doesn't increase and you should retrain the model.

you mean that just retrain it and do not change any super-parameters such as lr or batch-size? just re-run the same command "python train.py --opt options/train/base7.yaml --name base7_D4C28_bs16ps64_lr1e-2 --scale 2 --bs 16 --ps 64 --lr 1e-3 --gpu_ids 0"?

@NJU-Jet
Copy link
Owner

NJU-Jet commented Jun 1, 2021

Yes. In my experiment, simple retrain will solve this problem.

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

2 participants