提交 11354880 编写于 作者: K kolinwei

change some code style

上级 1d69b1ff
...@@ -184,8 +184,8 @@ def run_benchmark(model, args): ...@@ -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]))
......
...@@ -188,7 +188,9 @@ def main(): ...@@ -188,7 +188,9 @@ def main():
loss, acc, weight = train_exe.run( loss, acc, weight = train_exe.run(
feed={"pixel": img_data, feed={"pixel": img_data,
"label": y_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
])
accuracy.add(value=np.array(np.mean(acc)), weight=np.mean(weight)) accuracy.add(value=np.array(np.mean(acc)), weight=np.mean(weight))
iters += 1 iters += 1
num_samples += len(y_data) num_samples += len(y_data)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册