Fix bug of model saving

yzhangcs · web-flow · commit 716ac3ed35c5 · 2021-08-28T22:47:15.000+08:00
diff --git a/supar/parsers/parser.py b/supar/parsers/parser.py
@@ -91,8 +91,9 @@ def train(self, train, dev, test, buckets=32, batch_size=5000, update_steps=1,
                 logger.info(f"{t}s elapsed\n")
             if self.patience < 1:
                 break
-        loss, metric = self.load(**args)._evaluate(test.loader)
-        self.save(args.path)
+        parser = self.load(**args)
+        loss, metric = parser._evaluate(test.loader)
+        parser.save(args.path)
 
         logger.info(f"Epoch {self.best_e} saved")
         logger.info(f"{'dev:':5} {self.best_metric}")
