add load pre-trained model

Author: catqaq

main.py

@@ -154,7 +154,7 @@ def get_dataset(csv_data, text_field, label_field):
         test_acc.append(tmp)
         if tmp > max_acc:
             max_acc = tmp
-            #torch.save(model.state_dict(), './result/'+classifier+'.pt')
+            torch.save(model.state_dict(), './result/'+classifier+'.pt')
     print('Repeat %s times: %s' % (t, test_acc))
     print('average test_acc: %.1f%%' % (100*sum(test_acc)/t))
     print('max acc: %.1f%%' % (100*max_acc))
@@ -164,4 +164,4 @@ def get_dataset(csv_data, text_field, label_field):
     model.load_state_dict(torch.load('./result/'+classifier+'.pt'))
     model.eval()
     print(eval(test_iter, model, device))
-    
+