tests/test_zero/test_sharded_optim_v2.py · deaf99f4c98559edce9c3dc06d44e8a8e13c1eaa · OpenDAS / ColossalAI

"tests/test_zero/test_sharded_optim_v2.py" did not exist on "3af13a2c3e917ce23d44050cfeea71cfa9f23e81"

ver217 authored Apr 13, 2022

* add tensor placement policies

* polish comments

* polish comments

* update moe unit tests

e396bb71

test_sharded_optim_v2.py 4.68 KB