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
ver217 9ec1ce6ab1
[zero] sharded model support the reuse of fp16 shard (#495)
3 years ago
..
__init__.py [zero] polish sharded param name (#484) 3 years ago
sharded_param.py [zero] sharded model support the reuse of fp16 shard (#495) 3 years ago
sharded_tensor.py [zero] find miss code (#378) 3 years ago