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
ver217 9ec1ce6ab1
[zero] sharded model support the reuse of fp16 shard (#495)
3 years ago
..
init_ctx
shard_utils
sharded_model [zero] sharded model support the reuse of fp16 shard (#495) 3 years ago
sharded_optim [zero] sharded model support the reuse of fp16 shard (#495) 3 years ago
sharded_param [zero] sharded model support the reuse of fp16 shard (#495) 3 years ago
__init__.py