-
Notifications
You must be signed in to change notification settings - Fork 486
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
About model training #117
Comments
Hi there. Did you solve your problem? I had the same issue. How many images do u have for the training? |
I had the same issue. How many images do u have for the training? |
My dataset is only 1000 images, but after checking the code, I found that we should use random crop instead of using center crop (because only 1k images). In my case, If I train from scratch (4 GPUs A100), after 100k steps the generated images will be clearer. If I use a pre-trained model to train, after about 50k steps the generated would be better. |
Sorry for not replying in time. The ImageNet dataset I used for training has about 1.2 million images. After checking my code, I found that my problem was caused by a coding error in the sampling process. So I'm sorry I can't answer your question. |
Do I need to take any other steps during the training process? After training for a few days, the loss did not continue to decrease and the sampling results obtained were also very blurry. Do anyone have any training suggestions?
Here are some of my sampling results. I trained it for 250K iterations.
The text was updated successfully, but these errors were encountered: