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.
ColossalAI/colossalai/zero/sharded_param
Jiarui Fang e956d93ac2
[refactor] memory utils (#577)
3 years ago
..
__init__.py [zero] polish sharded param name (#484) 3 years ago
sharded_param.py [refactor] memory utils (#577) 3 years ago
sharded_tensor.py [zero] label state for param fp16 and grad (#551) 3 years ago
tensorful_state.py [zero] label state for param fp16 and grad (#551) 3 years ago