Commits · flax-community/dalle-mini

feat(train): update sweep config

bbbf7c8

boris commited on Jan 19, 2022

fix(train): variable not defined

4c87adf

boris commited on Jan 18, 2022

feat(train): cleanup args

a2bf605

boris commited on Jan 18, 2022

Merge pull request #122 from borisdayma/feat-acccum

c91ceb7
unverified

boris commited on Jan 17, 2022

feat(data): support accumulation in non-streaming

88c8e06

boris commited on Jan 17, 2022

refactor(train): cleanup

274ba73

boris commited on Jan 17, 2022

feat: custom gradient accumulation

2d07559

boris commited on Jan 17, 2022

fix: style

df01fa8

boris commited on Jan 16, 2022

feat(train): use MultiSteps for gradient accumulation

4fa53a5

boris commited on Jan 16, 2022

Merge pull request #118 from borisdayma/feat-optim

193c88c
unverified

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

fix(data): no shuffling of validation data

ddcbc6a

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

feat: support pypi

f5dba1e

boris commited on Jan 15, 2022

doc: update contributions

e3b1b56

boris commited on Jan 15, 2022

Merge pull request #117 from borisdayma/fix-inference

ef985be
unverified

boris commited on Jan 15, 2022

fix(inference): use float32 + flatten logits

71c4de3

boris commited on Jan 15, 2022

Merge pull request #115 from borisdayma/feat-shampoo

3a3d375
unverified

boris commited on Jan 14, 2022

feat: update inference pipeline

af807f7

boris commited on Jan 14, 2022

style: isort

531cd78

boris commited on Jan 14, 2022

style: apply to distributed_shampoo

e669c1b

boris commited on Jan 14, 2022

feat: add best_effort_memory_usage_reduction

4d518c7

boris commited on Jan 14, 2022

doc: add reference to Distributed Shampoo

db882b8

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