bug fix

2019-05-15 12:37:31 +02:00 · 2019-05-15 12:37:31 +02:00 · 832dc3eafa
parent 12046846d4
commit 832dc3eafa
1 changed files with 0 additions and 1 deletions
--- a/utils/generic_utils.py
+++ b/utils/generic_utils.py
@ -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,