Hugging Face
Models
Datasets
Spaces
Community
Docs
Enterprise
Pricing
Log In
Sign Up
ydshieh
/
flax-vision-encoder-decoder-vit-gpt2-coco-en
like
0
Model card
Files
Files and versions
xet
Community
0fc714d
flax-vision-encoder-decoder-vit-gpt2-coco-en
/
run_image_captioning_flax.py
Commit History
use custom TrainingArguments
0fc714d
ydshieh
commited on
Dec 16, 2021
use jax rng
9ca46fa
ydshieh
commited on
Dec 16, 2021
fix speed 2
a231b72
ydshieh
commited on
Dec 16, 2021
fix speed
efe62b9
ydshieh
commited on
Dec 16, 2021
more general
283180e
ydshieh
commited on
Dec 15, 2021
keep in memory for feature extraction
cb2b435
ydshieh
commited on
Dec 15, 2021
workaround for tokenizer
fde7d97
ydshieh
commited on
Dec 15, 2021
better epoch desc on terminal
36f2642
ydshieh
commited on
Dec 14, 2021
cleanup
c32f83d
ydshieh
commited on
Dec 14, 2021
cleanup
e70a466
ydshieh
commited on
Dec 14, 2021
fix
4450aac
ydshieh
commited on
Dec 14, 2021
separate tokenization and feature extraction
8f31d11
ydshieh
commited on
Dec 14, 2021
fix tokenization hang on TPU VM when worker > 1
9f6265f
ydshieh
commited on
Dec 13, 2021
cleanup
c2bb67b
ydshieh
commited on
Dec 13, 2021
remove incorrect comment
a552802
ydshieh
commited on
Dec 13, 2021
init
6984f2b
ydshieh
commited on
Dec 13, 2021