Hugging Face's logo

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

Model card Files
xet
Community
flax-vision-encoder-decoder-vit-gpt2-coco-en / run_image_captioning_flax.py

Commit History

improve doc
afddfdc

ydshieh commited on

improve doc
9a97c24

ydshieh commited on

fix
212ecaa

ydshieh commited on

fix
aebbb72

ydshieh commited on

fix desc
13c3593

ydshieh commited on

make block_size 0 work
eabb817

ydshieh commited on

imporve feature_extraction_fn
a897ce1

ydshieh commited on

update
89bf8e9

ydshieh commited on

update
6b61700

ydshieh commited on

fix epoch
0487060

ydshieh commited on

fix
5b183ad

ydshieh commited on

fix
93a6e2b

ydshieh commited on

fix
99c1534

ydshieh commited on

clean up
10c0ce9

ydshieh commited on

update
cff52cf

ydshieh commited on

use custom TrainingArguments
0fc714d

ydshieh commited on

use jax rng
9ca46fa

ydshieh commited on

fix speed 2
a231b72

ydshieh commited on

fix speed
efe62b9

ydshieh commited on

more general
283180e

ydshieh commited on

keep in memory for feature extraction
cb2b435

ydshieh commited on

workaround for tokenizer
fde7d97

ydshieh commited on

better epoch desc on terminal
36f2642

ydshieh commited on

cleanup
c32f83d

ydshieh commited on

cleanup
e70a466

ydshieh commited on

fix
4450aac

ydshieh commited on

separate tokenization and feature extraction
8f31d11

ydshieh commited on

fix tokenization hang on TPU VM when worker > 1
9f6265f

ydshieh commited on

cleanup
c2bb67b

ydshieh commited on

remove incorrect comment
a552802

ydshieh commited on

init
6984f2b

ydshieh commited on