Skip to content

Commit 1d69b1f

Browse files
committed
change some code style
1 parent 2b63b03 commit 1d69b1f

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

benchmark/fluid/mnist.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -179,9 +179,13 @@ def run_benchmark(model, args):
179179
outs = train_exe.run(
180180
feed={"pixel": img_data,
181181
"label": y_data},
182-
fetch_list=[avg_cost.name, batch_acc.name, batch_size_tensor.name]
182+
fetch_list=[
183+
avg_cost.name, batch_acc.name, batch_size_tensor.name
184+
]
183185
) # The accuracy is the accumulation of batches, but not the current batch.
184-
accuracy.update(value=np.array(np.mean(outs[1])), weight=np.mean(np.array(outs[2])))
186+
accuracy.update(
187+
value=np.array(np.mean(outs[1])),
188+
weight=np.mean(np.array(outs[2])))
185189
iters += 1
186190
num_samples += len(y_data)
187191
loss = np.mean(np.array(outs[0]))

benchmark/fluid/resnet.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,9 @@ def test(exe):
268268
loss, acc, weight = train_exe.run(
269269
feed={'data': image,
270270
'label': label},
271-
fetch_list=[avg_cost.name, batch_acc.name, batch_size_tensor.name])
271+
fetch_list=[
272+
avg_cost.name, batch_acc.name, batch_size_tensor.name
273+
])
272274
iters += 1
273275
num_samples += len(label)
274276
accuracy.add(value=np.array(np.mean(acc)), weight=np.mean(weight))

0 commit comments

Comments
 (0)