Merge pull request bristolcrypto#65 from Mazelt/ml-optimizer

Fix: out of range bug in optimizer.run
amiller · Feb 26, 2020 · 7d44986 · 7d44986
2 parents b17d4b6 + 3f0c76e
commit 7d44986
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Compiler/ml.py b/Compiler/ml.py
@@ -689,7 +689,7 @@ def _(j):
                         indices = indices_by_label[label]
                         @for_range_multithread(self.n_threads, 1, n)
                         def _(i):
-                            idx = indices[i + j * n_per_epoch]
+                            idx = indices[i + j * n]
                             self.layers[0].X[i + label * n] = X[idx]
                     self.forward(None)
                     self.backward()