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

[NFC] polish colossalai/cli/benchmark/utils.py code style (#4254)

Browse Source
This commit is contained in:
ocd_with_naming 2023-07-18 10:54:27 +08:00 committed by binmakeswell
parent dee1c96344
commit 85774f0c1f
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
colossalai/cli/benchmark/utils.py
View File

@ -1,10 +1,11 @@
import math
import time
from typing import Callable, Dict, List, Tuple
import torch
from colossalai.context import Config, ParallelMode
from colossalai.utils import MultiTimer
from colossalai.context import ParallelMode, Config
from typing import List, Dict, Tuple, Callable
def get_time_stamp() -> int: