ydshieh
/

flax-vision-encoder-decoder-vit-gpt2-coco-en

Model card Files Files and versions

ydshieh commited on Dec 17, 2021

Commit

0487060

·

1 Parent(s): 5b183ad

fix epoch

Files changed (1) hide show

run_image_captioning_flax.py +2 -2

run_image_captioning_flax.py CHANGED Viewed

@@ -1295,7 +1295,7 @@ def main():
                 # ======================== Evaluating ==============================
                 if training_args.do_eval and (training_args.eval_steps is not None and training_args.eval_steps > 0) and cur_step % training_args.eval_steps == 0:
                     evaluate(input_rng, eval_dataset)
-                    save_results(epoch + 1, cur_step)
             # ======================== Epoch End ==============================
@@ -1313,7 +1313,7 @@ def main():
             if training_args.do_eval and (training_args.eval_steps is None or training_args.eval_steps <= 0):
                 evaluate(input_rng, eval_dataset)
-                save_results(epoch + 1, cur_step)
     # ======================== Prediction loop ==============================

                 # ======================== Evaluating ==============================
                 if training_args.do_eval and (training_args.eval_steps is not None and training_args.eval_steps > 0) and cur_step % training_args.eval_steps == 0:
                     evaluate(input_rng, eval_dataset)
+                    save_results(epoch, cur_step)
             # ======================== Epoch End ==============================
             if training_args.do_eval and (training_args.eval_steps is None or training_args.eval_steps <= 0):
                 evaluate(input_rng, eval_dataset)
+                save_results(epoch, cur_step)
     # ======================== Prediction loop ==============================