Skip to content

Commit cce8795

Browse files
committed
fix dist test: fit a line
1 parent e7d44a2 commit cce8795

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

python/paddle/v2/fluid/tests/book_distribute/notest_dist_fit_a_line.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,10 +68,10 @@
6868
fluid.io.save_persistables(exe, "./fit_a_line.model/")
6969
fluid.io.load_persistables(exe, "./fit_a_line.model/")
7070
for data in train_reader():
71-
avg_loss_value, = exe.run(trainer_prog,
72-
feed=feeder.feed(data),
73-
fetch_list=[avg_cost])
74-
71+
avg_loss_value = exe.run(trainer_prog,
72+
feed=feeder.feed(data),
73+
fetch_list=[avg_cost])
74+
print("loss:" + str(avg_loss_value))
7575
if avg_loss_value[0] < 10.0:
7676
exit(0)
7777
exit(1)

0 commit comments

Comments
 (0)