Update model_arch.py

model_arch.py

@@ -18,11 +18,10 @@ class TextClassifierModel(nn.Module):
         return self.fc(embedded_activated)
 
 
-def load_state_dict(new_model, new_optimizer, trained_model, vocab):
+def load_state_dict(new_model, trained_model, vocab):
     num_class = 11
     vocab_size = len(vocab)
     embed_size = 300
     new_model.load_state_dict(trained_model['model_state_dict'])
-    new_optimizer.load_state_dict(trained_model['optimizer_state_dict'])
-    return new_model, new_optimizer
+    return new_model