Explorar o código

Fix bug in benchmark_averaging.py from master

Aleksandr Borzunov %!s(int64=4) %!d(string=hai) anos
pai
achega
3c92944c9d
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      benchmarks/benchmark_averaging.py

+ 1 - 1
benchmarks/benchmark_averaging.py

@@ -57,7 +57,7 @@ def benchmark_averaging(
         dht = hivemind.DHT(initial_peers=initial_peers, start=True)
         initial_bits = bin(index % num_groups)[2:].rjust(nbits, "0")
         averager = hivemind.averaging.DecentralizedAverager(
-            peer_tensors[i],
+            peer_tensors[index],
             dht,
             prefix="my_tensor",
             initial_group_bits=initial_bits,