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__.py | ||
_utils.py | ||
sharded_optim_v2.py |