mirror of https://github.com/hpcaitech/ColossalAI
d73c2b1d79
* change torch.Parameter to ColoParameter * fix post assignment for init context * polish * polish |
||
---|---|---|
.. | ||
__init__.py | ||
colo_init_context.py | ||
pipelinable.py | ||
utils.py |