github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-04 19:16:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Flake8 code restyle

Browse Source
This commit is contained in:
yuxuan-lou 2022-03-09 15:17:01 +08:00 committed by Frank Lee
parent af801cb4df
commit 3b88eb2259
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/engine/ophooks/_memtracer_ophook.py
View File

@ -14,7 +14,6 @@ import math
class MemTracerOpHook(BaseOpHook):
"""
Collect GPU memory usage information
:param warmup: This parameter indicates how many iterations to truncate before profiling, defaults to 50
:type warmup: int
:param refreshrate: This parameter decides the frequency of write file, defaults to 10