From 49715a78f066e9f3d5b4405ead58af9e59f4d70b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E3=82=A2=E3=83=9E=E3=83=87=E3=82=A6=E3=82=B9?= Date: Tue, 3 Jan 2023 18:23:11 +0800 Subject: [PATCH] [NFC] polish colossalai/cli/benchmark/benchmark.py code style (#2287) --- colossalai/cli/benchmark/benchmark.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/colossalai/cli/benchmark/benchmark.py b/colossalai/cli/benchmark/benchmark.py index 43632b150..f40f8f2f9 100644 --- a/colossalai/cli/benchmark/benchmark.py +++ b/colossalai/cli/benchmark/benchmark.py @@ -1,16 +1,17 @@ -import colossalai +from functools import partial +from typing import Dict, List + import click import torch.multiprocessing as mp -from functools import partial -from typing import List, Dict - +import colossalai +from colossalai.cli.benchmark.utils import find_all_configs, get_batch_data, profile_model from colossalai.context import Config from colossalai.context.random import reset_seeds from colossalai.core import global_context as gpc from colossalai.logging import disable_existing_loggers, get_dist_logger -from colossalai.utils import free_port, MultiTimer -from colossalai.cli.benchmark.utils import find_all_configs, profile_model, get_batch_data +from colossalai.utils import MultiTimer, free_port + from .models import MLP @@ -53,7 +54,7 @@ def run_dist_profiling(rank: int, world_size: int, port_list: List[int], config_ port_list (List[int]): a list of free ports for initializing distributed networks config_list (List[Dict]): a list of configuration hyperparams (Config): the hyperparameters given by the user - + """ # disable logging for clean output