github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-05 02:51:59 +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
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 math
import time import time
from typing import Callable, Dict, List, Tuple
import torch import torch
from colossalai.context import Config, ParallelMode
from colossalai.utils import MultiTimer from colossalai.utils import MultiTimer
from colossalai.context import ParallelMode, Config
from typing import List, Dict, Tuple, Callable
def get_time_stamp() -> int: def get_time_stamp() -> int: