mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-01 03:45:27 +00:00
* [Tensor] activation is an attr of ColoTensor * [Tensor] add optimizer * only detach parameters in context * polish code |
||
---|---|---|
.. | ||
_ops | ||
optim | ||
__init__.py | ||
colo_tensor.py | ||
op_wrapper.py | ||
spec.py | ||
utils.py |