Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
Spaces:
flax-community
/
dalle-mini
like
338
Running
App
Files
Files
Community
9
df01fa8
dalle-mini
/
tools
/
train
/
train.py
Commit History
fix: style
df01fa8
boris
commited on
Jan 16, 2022
feat(train): use MultiSteps for gradient accumulation
4fa53a5
boris
commited on
Jan 16, 2022
fix: style
25862e8
boris
commited on
Jan 16, 2022
feat: add more config of distributed_shampoo
89cf9ea
boris
commited on
Jan 16, 2022
feat(train): refactor learning rate params
e2781bc
boris
commited on
Jan 16, 2022
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