other small stuff

NikolayBlagoev · Mar 6, 2024 · b704036 · b704036
1 parent 8740dd6
commit b704036
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deccom/protocols/abstractprotocol.py b/deccom/protocols/abstractprotocol.py
@@ -97,7 +97,7 @@ async def _lower_start(self, p: Peer):
         return
 
     async def start(self, p: Peer):
-        await self._lower_start()
+        await self._lower_start(p)
         print("started")
         self.peer = p
         self.started = True

diff --git a/swarmtrainer.py b/swarmtrainer.py
@@ -113,7 +113,7 @@ def forward(self, x):
 
 optimizer = optim.SGD(net.parameters(), lr=learning_rate,
                       momentum=momentum)
-training = SwarmProtocol(int(argv[1]) % 3,net,optimizer,enumerate(train_loader))
+training = SwarmProtocol(int(argv[1]) % 3,net,optimizer,train_loader)
 training.set_lower(stream)
 me = TrainingNode( Peer(None, pub_key=argv[1]), training,"127.0.0.1", 10015 if argv[1] == "0" else None)
 print( "TCP", me.tcp_port)