Artem Chumachenko 3 년 전
부모
커밋
4ddc635245
2개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 1
      hivemind/optim/grad_averager.py
  2. 1 1
      tests/test_optimizer.py

+ 1 - 1
hivemind/optim/grad_averager.py

@@ -1,5 +1,5 @@
 import contextlib
-from typing import Any, Callable, Union, Iterable, Iterator, Optional, Sequence, Type, TypeVar
+from typing import Any, Callable, Iterable, Iterator, Optional, Sequence, Type, TypeVar, Union
 
 import torch
 

+ 1 - 1
tests/test_optimizer.py

@@ -13,8 +13,8 @@ import torch.nn.functional as F
 import hivemind
 from hivemind.averaging.control import AveragingStage
 from hivemind.optim.grad_averager import GradientAverager, GradientAveragerFactory
-from hivemind.optim.power_ef_averager import PowerEFGradientAverager
 from hivemind.optim.optimizer import Optimizer
+from hivemind.optim.power_ef_averager import PowerEFGradientAverager
 from hivemind.optim.progress_tracker import ProgressTracker
 from hivemind.optim.state_averager import TrainingStateAverager
 from hivemind.utils.crypto import RSAPrivateKey