Thanks to visit codestin.com
Credit goes to Github.com

Skip to content

Conversation

@CookedMorsel
Copy link
Contributor

  1. When saving the discriminator state at the end of each epoch, accelerator.unwrap_model() wasn't called (a larger file than neccessary was created, and it wasn't loadable)
  2. When continuing from a checkpoint dir with no latest.pth, the model sometimes tried to load a discriminator state into the main model weights

@lawrence-cj
Copy link
Collaborator

Great one. Thanks so much for noticing.

@lawrence-cj lawrence-cj merged commit 7840b88 into NVlabs:main Nov 24, 2025
1 check passed
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

Successfully merging this pull request may close these issues.

2 participants