Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / tools /train

Commit History

fix: check local TPU instances only

87fed1b

boris commited on Dec 24, 2021

feat: add config

3f13951

boris commited on Dec 24, 2021

feat(train): handle multi-hosts

5b533b5

boris commited on Dec 24, 2021

style

a6252c9

boris commited on Dec 24, 2021

feat: minor improvements

53dade7

boris commited on Dec 21, 2021

fix: update model name

61c93f2

boris commited on Dec 20, 2021

fix(train): update model name

b257ca8

boris commited on Dec 20, 2021

feat: log num_params

1f57ad7

boris commited on Dec 20, 2021

fix: adjust training script + dataloader

a96f4dc

boris commited on Dec 19, 2021

feat(sweep): update config

de21250

boris commited on Nov 30, 2021

style: use isort

d209547

boris commited on Nov 30, 2021

feat(train): merge logged dict

baa52db

boris commited on Nov 30, 2021

chore: move files

46f7469

boris commited on Nov 30, 2021