diff --git a/benchmark/fluid/mnist.py b/benchmark/fluid/mnist.py index 1cb4314fb22ae935a00ceea25128e7f2b5430c84..8d51d4aa60dbfc201121a44cfd99cb9d5baa95ec 100644 --- a/benchmark/fluid/mnist.py +++ b/benchmark/fluid/mnist.py @@ -184,8 +184,8 @@ def run_benchmark(model, args): ] ) # The accuracy is the accumulation of batches, but not the current batch. accuracy.update( - value=np.array(np.mean(outs[1])), - weight=np.mean(np.array(outs[2]))) + value=np.array(np.mean(outs[1])), + weight=np.mean(np.array(outs[2]))) iters += 1 num_samples += len(y_data) loss = np.mean(np.array(outs[0]))