Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / tools /train /distributed_shampoo.py

Commit History

fix: style

d483294

boris commited on Feb 3, 2022

fix: distributed shampoo class

696422e

boris commited on Feb 2, 2022

feat: update distributed_shampoo

5996680

boris commited on Feb 2, 2022

feat: update distributed_shampoo + fix None spec

8a9e367

boris commited on Jan 24, 2022

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