github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 12:01:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

[polish] polish ColoTensor and its submodules (#2537)

Browse Source
This commit is contained in:
HELSON
2023-02-03 11:44:10 +08:00
committed by GitHub
parent 51d4d6e718
commit 552183bb74
6 changed files with 75 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/tensor/colo_parameter.py
View File

@@ -71,7 +71,7 @@ class ColoParameter(ColoTensor, torch.nn.Parameter):
return tensor
def __repr__(self):
return f'ColoParameter: {ColoTensor.__repr__(self)}'
return super(ColoParameter, self).__repr__()
@classmethod
def __torch_function__(cls, func, types, args=..., kwargs=None):