fix

run_image_captioning_flax.py

@@ -1177,7 +1177,7 @@ def main():
 
     if training_args.do_train:
         logger.info("***** Running training *****")
-        logger.info(f"  Num train examples = {len(train_dataset)}")
+        logger.info(f"  Num train examples = {num_train_examples_per_epoch}")
         logger.info(f"  Num Epochs = {num_epochs}")
         logger.info(f"  Instantaneous train batch size per device = {training_args.per_device_train_batch_size}")
         logger.info(f"  Total train batch size (w. parallel & distributed) = {train_batch_size}")