From 832dc3eafa41fdf677bfad0db91bb75ec1c989bc Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Wed, 15 May 2019 12:37:31 +0200
Subject: [PATCH] bug fix

---
 utils/generic_utils.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/utils/generic_utils.py b/utils/generic_utils.py
index 902affba..760cbae3 100644
--- 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,