mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-03 03:31:05 +06:00
fixed to find best dev acc
This commit is contained in:
parent
b57bfb5fa0
commit
8960988f35
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user