Artem Chumachenko преди 3 години
родител
ревизия
d7154f0c2f
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      hivemind/optim/power_sgd_averager.py

+ 1 - 1
hivemind/optim/power_sgd_averager.py

@@ -19,9 +19,9 @@ from hivemind.compression import (
     CompressionBase,
     CompressionInfo,
     NoCompression,
+    TensorRole,
     deserialize_torch_tensor,
     serialize_torch_tensor,
-    TensorRole,
 )
 from hivemind.dht import DHT, DHTID
 from hivemind.p2p import P2P, P2PContext, P2PHandlerError, PeerID, ServicerBase