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

Mode collapse with a large dataset #104

Closed
Poofy1 opened this issue May 7, 2021 · 3 comments
Closed

Mode collapse with a large dataset #104

Poofy1 opened this issue May 7, 2021 · 3 comments

Comments

@Poofy1
Copy link

Poofy1 commented May 7, 2021

First off I would just like to say this most likely is not a bug. I am mostly just asking for advice because I'm not very familiar with the program's config and this page didn't explain the settings in much depth.

I have a dataset of about 65000 images and I expected the snapshot images to be very diverse but they are all very similar. This started becoming noticeable around the 200kimg mark. I believe this is mode collapse. I am using the default settings currently. I have not really tried anything because I don't know what to change. Perhaps increasing batch_size or ada_kimg will help prevent it?

My main questions:
Is there a config setting that I can change that will help prevent mode collapse from happening? Do you have any other advice to prevent mode collapse?

fakes000240

Desktop:

  • OS: Windows 10
  • PyTorch version 1.7.1
  • CUDA toolkit version CUDA 11.1
  • NVIDIA driver version: 466.11
  • RTX 3070
  • Anaconda enviroment
@abstractdonut
Copy link

Have you continued training your model since posting this? If so, do recent snapshots show more diversity?

Does your dataset have a lot of variation? Are all images cropped and centered and do they contain a single class of object from a single point of view, or are there multiple types of objects depicted?

@Poofy1
Copy link
Author

Poofy1 commented May 24, 2021

As I trained more it became more diverse, so it has seemed to fix itself. Thanks :)

@Poofy1 Poofy1 closed this as completed May 24, 2021
@katie-cathy-hunt
Copy link

@Poofy1 Hi! After how many kimgs did the snapshot become more diverse?

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