optimizer / test /test_muon /run_test.sh
wyldecat's picture
feat(muon): add test for muon
3261444
raw
history blame contribute delete
54 Bytes
NGPU=${NGPU:-"8"}
torchrun --nproc-per-node=8 test.py