dalle-mini / tools /train /distributed_shampoo.py

Commit History

fix: distributed shampoo class
696422e

boris commited on

feat: update distributed_shampoo
5996680

boris commited on

feat: update distributed_shampoo + fix None spec
8a9e367

boris commited on

style: isort
531cd78

boris commited on

style: apply to distributed_shampoo
e669c1b

boris commited on

feat: update distributed_shampoo
b90198c

boris commited on

feat: add shampoo optimizer
0b87452

boris commited on