|
|
@ -184,8 +184,8 @@ def run_benchmark(model, args):
|
|
|
|
]
|
|
|
|
]
|
|
|
|
) # The accuracy is the accumulation of batches, but not the current batch.
|
|
|
|
) # The accuracy is the accumulation of batches, but not the current batch.
|
|
|
|
accuracy.update(
|
|
|
|
accuracy.update(
|
|
|
|
value=np.array(np.mean(outs[1])),
|
|
|
|
value=np.array(np.mean(outs[1])),
|
|
|
|
weight=np.mean(np.array(outs[2])))
|
|
|
|
weight=np.mean(np.array(outs[2])))
|
|
|
|
iters += 1
|
|
|
|
iters += 1
|
|
|
|
num_samples += len(y_data)
|
|
|
|
num_samples += len(y_data)
|
|
|
|
loss = np.mean(np.array(outs[0]))
|
|
|
|
loss = np.mean(np.array(outs[0]))
|
|
|
|