未验证 提交 a08389e9 编写于 作者: S Steffy-zxf 提交者: GitHub

Update autofinetune-nlp.md

上级 f73f82ad
......@@ -133,8 +133,9 @@ if __name__ == '__main__':
eval_avg_score, eval_avg_loss, eval_run_speed = cls_task._calculate_metrics(run_states)
# Move ckpt/best_model to the defined saved parameters directory
if is_path_valid(args.saved_params_dir) and os.path.exists(config.checkpoint_dir+"/best_model/"):
shutil.copytree(config.checkpoint_dir+"/best_model/", args.saved_params_dir)
best_model_dir = os.path.join(config.checkpoint_dir, "best_model")
if is_path_valid(args.saved_params_dir) and os.path.exists(best_model_dir):
shutil.copytree(best_model_dir, args.saved_params_dir)
shutil.rmtree(config.checkpoint_dir)
print("AutoFinetuneEval"+"\t"+str(float(eval_avg_score["acc"])))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册