Fix bug in logistic regression benchmark.

This commit is contained in:
Marcel Keller
2022-06-23 19:17:08 +02:00
parent ec1d302b03
commit af5af2df25

View File

@@ -9,6 +9,7 @@ cfix.set_precision(16, 31)
dim = int(program.args[1])
batch = int(program.args[2])
ml.Layer.back_batch_size = batch
try:
n_iterations = int(program.args[3])