github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-07 16:08:22 +00:00
Code Issues Packages Projects Releases Wiki Activity
fce9432f08
ColossalAI/tests/components_to_test
History
Jiarui Fang 370f567e7d
[zero] new interface for ShardedOptimv2 (#406)
2022-03-14 20:48:41 +08:00
..
utils [unit test] Refactored test cases with component func (#339) 2022-03-11 15:50:28 +08:00
__init__.py add bert for unitest and sharded model is not able to pass the bert case 2022-03-11 15:50:28 +08:00
bert.py [zero] new interface for ShardedOptimv2 (#406) 2022-03-14 20:48:41 +08:00
nested_model.py [zero] new interface for ShardedOptimv2 (#406) 2022-03-14 20:48:41 +08:00
registry.py [test] refactored testing components (#324) 2022-03-11 15:50:28 +08:00
repeated_computed_layer.py [zero] new interface for ShardedOptimv2 (#406) 2022-03-14 20:48:41 +08:00
resnet.py [zero] new interface for ShardedOptimv2 (#406) 2022-03-14 20:48:41 +08:00