File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -184,8 +184,8 @@ def run_benchmark(model, args):
184
184
]
185
185
) # The accuracy is the accumulation of batches, but not the current batch.
186
186
accuracy .update (
187
- value = np .array (np .mean (outs [1 ])),
188
- weight = np .mean (np .array (outs [2 ])))
187
+ value = np .array (np .mean (outs [1 ])),
188
+ weight = np .mean (np .array (outs [2 ])))
189
189
iters += 1
190
190
num_samples += len (y_data )
191
191
loss = np .mean (np .array (outs [0 ]))
Original file line number Diff line number Diff line change @@ -188,7 +188,9 @@ def test(exe):
188
188
loss , acc , weight = train_exe .run (
189
189
feed = {"pixel" : img_data ,
190
190
"label" : y_data },
191
- fetch_list = [avg_cost .name , batch_acc .name , batch_size_tensor .name ])
191
+ fetch_list = [
192
+ avg_cost .name , batch_acc .name , batch_size_tensor .name
193
+ ])
192
194
accuracy .add (value = np .array (np .mean (acc )), weight = np .mean (weight ))
193
195
iters += 1
194
196
num_samples += len (y_data )
You can’t perform that action at this time.
0 commit comments