Fixing Saving Error Issue

Author: codertimo

bert_pytorch/train.py

@@ -61,7 +61,7 @@ def train():
     print("Training Start")
     for epoch in range(args.epochs):
         trainer.train(epoch)
-        trainer.save(args.output_path, epoch)
+        trainer.save(epoch, args.output_path)
 
         if test_data_loader is not None:
             trainer.test(epoch)