소스 검색

Fix bug in benchmark_averaging.py from master

Aleksandr Borzunov 4 년 전
부모
커밋
3c92944c9d
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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,