mirror of https://github.com/hpcaitech/ColossalAI
3c9cd5bb5e
* [WIP] stateful tensor manager * add eviction strategy * polish code * polish code * polish comment * add unit test * fix sampler bug * polish code * fix max sampling cnt resetting bug * fix sampler bug * polish code * fix bug * fix unit test Co-authored-by: jiaruifang <fangjiarui123@gmail.com> |
||
---|---|---|
.. | ||
init_ctx | ||
shard_utils | ||
sharded_model | ||
sharded_optim | ||
sharded_param | ||
__init__.py |