Hugging Face's logo

Spaces:
flax-community
/
dalle-mini
Running

App Files Community
9
dalle-mini / tools /train /train.py

Commit History

feat(train): handle multi-hosts
5b533b5

boris commited on

style
a6252c9

boris commited on

feat: minor improvements
53dade7

boris commited on

fix: update model name
61c93f2

boris commited on

fix(train): update model name
b257ca8

boris commited on

feat: log num_params
1f57ad7

boris commited on

fix: adjust training script + dataloader
a96f4dc

boris commited on

style: use isort
d209547

boris commited on

feat(train): merge logged dict
baa52db

boris commited on

chore: move files
46f7469

boris commited on