소스 검색

fix some issues in pr

xtinkt 4 년 전
부모
커밋
15f1e9b0b5
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      hivemind/optim/__init__.py

+ 1 - 1
hivemind/optim/__init__.py

@@ -1,5 +1,5 @@
 from hivemind.optim.base import DecentralizedOptimizerBase
 from hivemind.optim.collaborative import CollaborativeOptimizer
 from hivemind.optim.performance_ema import PerformanceEMA
-from hivemind.optim.decentralized_optimizers import DecentralizedOptimizer, DecentralizedSGD, DecentralizedAdam
+from hivemind.optim.averaged import DecentralizedOptimizer, DecentralizedSGD, DecentralizedAdam
 from hivemind.optim.adaptive import CollaborativeAdaptiveOptimizer