提交 1d69b1ff 编写于 作者: K kolinwei

change some code style

上级 2b63b034
......@@ -179,9 +179,13 @@ def run_benchmark(model, args):
outs = train_exe.run(
feed={"pixel": img_data,
"label": y_data},
fetch_list=[avg_cost.name, batch_acc.name, batch_size_tensor.name]
fetch_list=[
avg_cost.name, batch_acc.name, batch_size_tensor.name
]
) # 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])))
accuracy.update(
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]))
......
......@@ -268,7 +268,9 @@ def run_benchmark(model, args):
loss, acc, weight = train_exe.run(
feed={'data': image,
'label': label},
fetch_list=[avg_cost.name, batch_acc.name, batch_size_tensor.name])
fetch_list=[
avg_cost.name, batch_acc.name, batch_size_tensor.name
])
iters += 1
num_samples += len(label)
accuracy.add(value=np.array(np.mean(acc)), weight=np.mean(weight))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册