Jiarui Fang 8daf1b4db1 [Gemini] patch for supporting orch.add_ function for ColoTensor (#2003)		2022-11-25 20:06:35 +08:00
..
utils	[Gemini] add unitests to check gemini correctness (#2015)	2022-11-24 16:51:45 +08:00
__init__.py	[Gemini] add unitests to check gemini correctness (#2015)	2022-11-24 16:51:45 +08:00
bert.py	[Tensor] fix test_model (#916)	2022-05-06 18:06:22 +08:00
gpt.py	[Gemini] add unitests to check gemini correctness (#2015)	2022-11-24 16:51:45 +08:00
inline_op_model.py	[Gemini] patch for supporting orch.add_ function for ColoTensor (#2003)	2022-11-25 20:06:35 +08:00
nested_model.py	[zero] new interface for ShardedOptimv2 (#406)	2022-03-14 20:48:41 +08:00
no_leaf_module.py	[zero] adapt zero for unsharded paramters (Optimizer part) (#601)	2022-04-01 20:10:47 +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
simple_net.py	[Tensor] add 1d vocab loss (#918)	2022-05-07 15:49:14 +08:00