fixed to find best dev acc

This commit is contained in:
erenup 2019-09-19 01:10:05 +08:00
parent b57bfb5fa0
commit 8960988f35

View File

@ -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