diff --git a/benchmark/fluid/mnist.py b/benchmark/fluid/mnist.py index 8d51d4aa60dbfc201121a44cfd99cb9d5baa95ec..400200c4745017bd9d160bb9e415fde041c0a6c8 100644 --- a/benchmark/fluid/mnist.py +++ b/benchmark/fluid/mnist.py @@ -184,7 +184,7 @@ 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])), + value=np.array(np.mean(outs[1])), weight=np.mean(np.array(outs[2]))) iters += 1 num_samples += len(y_data)