ColossalAI/tests/test_engine
Jiarui Fang 370f567e7d
[zero] new interface for ShardedOptimv2 (#406)
2022-03-14 20:48:41 +08:00
..
test_engine.py [zero] new interface for ShardedOptimv2 (#406) 2022-03-14 20:48:41 +08:00
test_param_hook.py [test] refactored testing components (#324) 2022-03-11 15:50:28 +08:00