Skip to content

Commit a49c6db

Browse files
authored
remove unused import (#261)
1 parent f5888e9 commit a49c6db

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

torchft/local_sgd.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,18 +10,14 @@
1010
"""
1111
import logging
1212
import math
13-
import threading
1413
from contextlib import nullcontext
1514
from types import TracebackType
16-
from typing import Any, Callable, Dict, Iterator, List, Optional, Tuple, Type
15+
from typing import Any, Dict, List, Optional, Tuple, Type
1716

1817
import torch
19-
import torch.distributed as dist
2018
from torch import nn, optim
2119
from torch.distributed.distributed_c10d import Work
2220
from torch.distributed.tensor import DTensor
23-
from torch.nn.parameter import Parameter
24-
from torch.optim.optimizer import Optimizer
2521
from torch.utils.hooks import RemovableHandle
2622

2723
from torchft.manager import Manager

0 commit comments

Comments
 (0)