Explorar el Código

fix some issues in pr

xtinkt hace 4 años
padre
commit
15f1e9b0b5
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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