mirror of
https://github.com/wassname/TTS.git
synced 2026-10-20 11:59:38 +08:00
bug fix
This commit is contained in:
1 parent
12046846d4
commit
832dc3eafa
1 file changed
-1
@@ -115,7 +115,6 @@ def save_checkpoint(model, optimizer, optimizer_st, model_loss, out_path,
|
||||
state = {
|
||||
'model': new_state_dict,
|
||||
'optimizer': optimizer.state_dict(),
|
||||
'optimizer_st': optimizer_st.state_dict(),
|
||||
'step': current_step,
|
||||
'epoch': epoch,
|
||||
'linear_loss': model_loss,
|
||||
|
||||
Reference in new issue
Block a user