kennethsinder/oct-opus

When resetting the model, also reset the optimizer weights

Closed this issue · 0 comments

Currently, ModelState.reset_weights() resets only the discriminator and generator weights.
We should also be resetting the discriminator optimizer and the generator optimizer.