_flax-vision-encoder-decoder-vit-gpt2-coco-en / run_image_captioning_flax.py

Commit History

add eval / pred back
bf405d2

ydshieh commited on

Fix shape
fee9810

ydshieh commited on

update train script - part 1
787cca2

ydshieh commited on

update training script
56b6406

ydshieh commited on

Fix
57d6fb8

ydshieh commited on

Fix
324e75a

ydshieh commited on

Save at certain steps
7c2d026

ydshieh commited on

update scripts
c6809fe

ydshieh commited on

rename script
5370d02

ydshieh commited on