search

coqui-ai / Trainer

🐸 - A general purpose model trainer, as flexible as it gets
185 stars 109 forks source link

fix bug in restore_model that crashes when loading existing GAN checkpoints from Coqui TTS #101

Closed shenberg closed 1 year ago

shenberg commented 1 year ago

The bug is triggered when the field 'optimizer' in the checkpoint dictionary is a list, such as with GAN models where it's a list of 2 states, one for D optimizer and one for G optimizer.

erogol commented 1 year ago

Thanks for the PR. Apparently It is already fixed.

shenberg commented 1 year ago

Great!