From 8a0247573c7daa39841a01c6859e6a2469d70972 Mon Sep 17 00:00:00 2001 From: Giuseppe Romagnuolo Date: Wed, 7 Jun 2023 21:24:02 +0100 Subject: [PATCH] Fix: validation step incorrectly used training dataset, change to validation dataset --- docs/tutorials/tfr_bert.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/tutorials/tfr_bert.ipynb b/docs/tutorials/tfr_bert.ipynb index 804a368..9bc098c 100644 --- a/docs/tutorials/tfr_bert.ipynb +++ b/docs/tutorials/tfr_bert.ipynb @@ -356,7 +356,7 @@ " print(\"Training metrics for epoch: \" + str(train_step) + \" \", train_metrics)\n", "\n", " if train_step % EVAL_STEPS == 0:\n", - " task.validation_step(next(train_iterator), model, metrics=metrics)\n", + " task.validation_step(next(vali_iterator), model, metrics=metrics)\n", " vali_metrics = {m.name: m.result().numpy() for m in metrics}\n", " print(\"Validation metrics for epoch: \" + str(train_step) + \" \",\n", " vali_metrics)"