You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

ver217 67c33f57eb [tensor] design DistSpec and DistSpecManager for ColoTensor (#934) ... * add dist spec * update linear op * polish code * polish code * update embedding op * polish unit tests * polish unit tests * polish comments * polish code * add test_dist_spec_mgr * polish code * refactor folder structure * polish unit tests * add get_process_group() for TensorSpec * polish code		3 years ago
..
_ops	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	3 years ago
graph	[Graph] building computing graph with ColoTensor, Linear only (#917)	3 years ago
optim	[Tensor] initialize the ColoOptimizer (#898)	3 years ago
__init__.py	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	3 years ago
colo_parameter.py	[Tensor] init ColoParameter (#914)	3 years ago
colo_tensor.py	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	3 years ago
const.py	[Tensor] init ColoParameter (#914)	3 years ago
dist_spec.py	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	3 years ago
dist_spec_mgr.py	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	3 years ago
op_wrapper.py	…
spec.py	[tensor] design DistSpec and DistSpecManager for ColoTensor (#934)	3 years ago
utils.py	[Tensor] get named parameters for model using ColoTensors (#874)	3 years ago