Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / tools /train /distributed_shampoo.py

Commit History

style: isort

531cd78

boris commited on Jan 14, 2022

style: apply to distributed_shampoo

e669c1b

boris commited on Jan 14, 2022

feat: update distributed_shampoo

b90198c

boris commited on Jan 13, 2022

feat: add shampoo optimizer

0b87452

boris commited on Jan 8, 2022