Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / tools /train

Commit History

feat: add best_effort_memory_usage_reduction

4d518c7

boris commited on Jan 14, 2022

feat: update distributed_shampoo

b90198c

boris commited on Jan 13, 2022

fix: weight decay Adam + speed logging

7143593

boris commited on Jan 12, 2022

feat: add micro config

e501f71

boris commited on Jan 10, 2022

fix: shampoo -> distributed shampoo

edae62d

boris commited on Jan 9, 2022

feat: update params

604a65d

boris commited on Jan 9, 2022

feat: add shampoo optimizer

0b87452

boris commited on Jan 8, 2022

feat: update sweep

e1555d4

boris commited on Jan 6, 2022

feat: create config files

dc5c024

boris commited on Jan 6, 2022

feat: allow abstract_init

772415c

boris commited on Jan 5, 2022

fix: typo

5c84978

boris commited on Jan 5, 2022

feat: log more metrics

1b757dc

boris commited on Jan 5, 2022

feat: shard by host is optional

901ff72

boris commited on Jan 3, 2022

feat: load data first

fdf7698

boris commited on Dec 24, 2021

feat: display local TPU's

15993e3

boris commited on Dec 24, 2021

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