Merge pull request #527 from Mathieu-Prouveur/fix_value_training_loss

Update example files so that tr_loss is not affected by args.gradient…
This commit is contained in:
Thomas Wolf authored and GitHub committed 2019-04-30 11:12:55 +02:00
commit 2dee86319d
2 files changed
+3 -3

No files matched your search

+2 -2
View File
@@ -939,7 +939,7 @@ def main():
elif output_mode == "regression":
preds = np.squeeze(preds)
result = compute_metrics(task_name, preds, all_label_ids.numpy())
loss = tr_loss/nb_tr_steps if args.do_train else None
loss = tr_loss/global_step if args.do_train else None
result['eval_loss'] = eval_loss
result['global_step'] = global_step
@@ -1007,7 +1007,7 @@ def main():
preds = preds[0]
preds = np.argmax(preds, axis=1)
result = compute_metrics(task_name, preds, all_label_ids.numpy())
loss = tr_loss/nb_tr_steps if args.do_train else None
loss = tr_loss/global_step if args.do_train else None
result['eval_loss'] = eval_loss
result['global_step'] = global_step