Fix

PaddlePaddle · Oct 19, 2023 · 15bfc2f · 15bfc2f
1 parent bb5697e
commit 15bfc2f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/phylstm/phylstm2.py b/examples/phylstm/phylstm2.py
@@ -299,7 +299,7 @@ def evaluate(cfg: DictConfig):
         output_dir=cfg.output_dir,
         seed=cfg.seed,
         validator=validator_pde,
-        checkpoint_path=cfg.TRAIN.checkpoint_path,
+        checkpoint_path=cfg.EVAL.pretrained_model_path,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
     )
     # evaluate after finished training

diff --git a/examples/phylstm/phylstm3.py b/examples/phylstm/phylstm3.py
@@ -317,7 +317,7 @@ def evaluate(cfg: DictConfig):
         output_dir=cfg.output_dir,
         seed=cfg.seed,
         validator=validator_pde,
-        checkpoint_path=cfg.TRAIN.checkpoint_path,
+        checkpoint_path=cfg.EVAL.pretrained_model_path,
         eval_with_no_grad=cfg.EVAL.eval_with_no_grad,
     )