Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / src /dalle_mini /data.py

Commit History

feat(train): improve pjit speed

f254058

boris commited on Jan 25, 2022

fix(train): consider correct batch size

b7c7458

boris commited on Jan 25, 2022

style: unsused import

7a176b9

boris commited on Jan 23, 2022

feat(train): progress on pjit

49597a2

boris commited on Jan 22, 2022

feat(data): support accumulation in non-streaming

88c8e06

boris commited on Jan 17, 2022

feat: custom gradient accumulation

2d07559

boris commited on Jan 17, 2022

fix(data): no shuffling of validation data

ddcbc6a

boris commited on Jan 16, 2022

fix(train): handle seed_dataset

8b72ed8

boris commited on Jan 16, 2022

feat: support pypi

f5dba1e

boris commited on Jan 15, 2022