"git@developer.sourcefind.cn:Fzc7075/nunchaku.git" did not exist on "c78c99f691894ffff4102731f0562ca24778e2af"
Commit 97c30b73 authored by erenup's avatar erenup
Browse files

add test related code

parent d5e60e5b
......@@ -256,7 +256,7 @@ def evaluate(args, model, tokenizer, prefix="", test=False):
result = {"eval_acc": acc, "eval_loss": eval_loss}
results.update(result)
output_eval_file = os.path.join(eval_output_dir, "eval_results.txt")
output_eval_file = os.path.join(eval_output_dir, "is_test_" + str(test) + "_eval_results.txt")
with open(output_eval_file, "w") as writer:
logger.info("***** Eval results {} *****".format(str(prefix) + " is test:" + str(test)))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment