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

[NFC] polish colossalai/cli/benchmark/models.py code style (#3290)

Browse Source
This commit is contained in:
Ziheng Qin
2023-03-28 11:36:11 -07:00
committed by binmakeswell
parent 964a28678f
commit 1bed38ef37
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/cli/benchmark/models.py
View File

@@ -1,4 +1,5 @@
import torch
import colossalai.nn as col_nn