When resetting the model, also reset the optimizer weights
Closed this issue · 0 comments
Kaythay commented
Currently, ModelState.reset_weights()
resets only the discriminator and generator weights.
We should also be resetting the discriminator optimizer and the generator optimizer.