Commit 8960988f authored by erenup's avatar erenup
Browse files

fixed to find best dev acc

parent b57bfb5f
......@@ -170,7 +170,7 @@ def train(args, train_dataset, model, tokenizer):
results = evaluate(args, model, tokenizer)
for key, value in results.items():
tb_writer.add_scalar('eval_{}'.format(key), value, global_step)
if results["eval_acc"] < best_dev_acc:
if results["eval_acc"] > best_dev_acc:
best_dev_acc = results["eval_acc"]
best_dev_loss = results["eval_loss"]
best_steps = global_step
......
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