Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / tools /train /train.py

Commit History

fix(train): handle seed_dataset

8b72ed8

boris commited on Jan 16, 2022

feat: refactor TrainingArguments

adbdff9

boris commited on Jan 16, 2022

fix: push_to_hub deprecated

23389f6

boris commited on Jan 15, 2022

style: isort

531cd78

boris commited on Jan 14, 2022

feat: add best_effort_memory_usage_reduction

4d518c7

boris commited on Jan 14, 2022

fix: weight decay Adam + speed logging

7143593

boris commited on Jan 12, 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: 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(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

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