minor fix

Author: dhlee347

config/train_mrpc.json

@@ -1,9 +1,9 @@
 {
-    "seed": 3431,
+    "seed": 42,
     "batch_size": 32,
-    "lr": 5e-5,
-    "n_epochs": 15,
+    "lr": 2e-5,
+    "n_epochs": 3,
     "warmup": 0.1,
-    "save_steps": 50,
-    "total_steps": 10000
+    "save_steps": 100,
+    "total_steps": 345
 }

train.py

@@ -122,5 +122,5 @@ def load(self, model_file, pretrain_file):
     def save(self, e, i):
         """ save current model """
         torch.save(self.model.state_dict(), # save model object before nn.DataParallel
-            os.path.join(self.save_dir, 'model_epoch_'+str(e)+'_steps_'+str(i)+'.pt'))
+            os.path.join(self.save_dir, 'model_steps_'+str(i)+'.pt'))