Update model_arch.py

model_arch.py

@@ -16,3 +16,13 @@ class TextClassifierModel(nn.Module):
         embedded_activated = F.relu(embedded_norm)
 
         return self.fc(embedded_activated)
+
+
+def load_state_dict(new_model, new_optimizer, 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
+