ydshieh
/

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

Model card Files Files and versions

ydshieh commited on Dec 17, 2021

Commit

99c1534

·

1 Parent(s): 10c0ce9

fix

Files changed (1) hide show

run_image_captioning_flax.py +1 -1

run_image_captioning_flax.py CHANGED Viewed

@@ -1250,7 +1250,7 @@ def main():
                     json.dump(generations, fp, ensure_ascii=False, indent=4)
     def evaluate(rng, dataset):
-        evaluation_loop(rng, dataset, split='eval')
     def predict(rng, dataset):
         evaluation_loop(rng, dataset, split='test')

                     json.dump(generations, fp, ensure_ascii=False, indent=4)
     def evaluate(rng, dataset):
+        evaluation_loop(rng, dataset, split='valid')
     def predict(rng, dataset):
         evaluation_loop(rng, dataset, split='test')