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