github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-18 19:58:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[cli] remove unused imports (#1001)

Browse Source
This commit is contained in:
Frank Lee 2022-05-18 23:27:18 +08:00 committed by GitHub
parent 533d0c46d8
commit 1467d83edf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,4 +1,3 @@
from random import choices
import click import click
from .utils import * from .utils import *

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

@ -1,10 +1,8 @@
import math import math
import time import time
from grpc import Call
import torch import torch
from colossalai.utils import MultiTimer from colossalai.utils import MultiTimer
from colossalai.core import global_context as gpc
from colossalai.context import ParallelMode, Config from colossalai.context import ParallelMode, Config
from typing import List, Dict, Tuple, Callable from typing import List, Dict, Tuple, Callable

1
colossalai/cli/launcher/multinode_runner.py
View File

@ -1,5 +1,4 @@
import fabric import fabric
from fabric import Connection
from .hostinfo import HostInfo, HostInfoList from .hostinfo import HostInfo, HostInfoList
from multiprocessing import Pipe, Process from multiprocessing import Pipe, Process
from multiprocessing import connection as mp_connection from multiprocessing import connection as mp_connection