From 267867e8517cfcd6d930d8b71814a8cee40c64a3 Mon Sep 17 00:00:00 2001 From: Matt Date: Mon, 15 Nov 2021 13:45:51 +0000 Subject: [PATCH] Quick fix to TF summarization example (#14401) --- examples/tensorflow/summarization/run_summarization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/tensorflow/summarization/run_summarization.py b/examples/tensorflow/summarization/run_summarization.py index 9fda13cd89f..480235ad669 100644 --- a/examples/tensorflow/summarization/run_summarization.py +++ b/examples/tensorflow/summarization/run_summarization.py @@ -577,7 +577,7 @@ def main(): # region Optimizer, loss and LR scheduling # Scheduler and math around the number of training steps. - num_update_steps_per_epoch = len(train_dataset) // training_args.per_device_train_batch_size + num_update_steps_per_epoch = len(train_dataset) // total_train_batch_size num_train_steps = training_args.num_train_epochs * num_update_steps_per_epoch optimizer, lr_schedule = create_optimizer( init_lr=training_args.learning_rate, num_train_steps=num_train_steps, num_warmup_steps=0