justheuristic 4 жил өмнө
parent
commit
36abd5bdc1

+ 1 - 0
hivemind/__init__.py

@@ -17,6 +17,7 @@ from hivemind.optim import (
     DecentralizedOptimizerBase,
     DecentralizedOptimizerBase,
     DecentralizedSGD,
     DecentralizedSGD,
     TrainingAverager,
     TrainingAverager,
+    Optimizer
 )
 )
 from hivemind.p2p import P2P, P2PContext, P2PHandlerError, PeerID, PeerInfo
 from hivemind.p2p import P2P, P2PContext, P2PHandlerError, PeerID, PeerInfo
 from hivemind.utils import *
 from hivemind.utils import *