Skip to content

Commit 917b10b

Browse files
authored
Merge pull request #7799 from putcn/develop
add output to fit a line demo
2 parents e7d44a2 + d58923d commit 917b10b

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

python/paddle/v2/fluid/tests/book/test_fit_a_line.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@
4949
avg_loss_value, = exe.run(fluid.default_main_program(),
5050
feed=feeder.feed(data),
5151
fetch_list=[avg_cost])
52-
52+
print(avg_loss_value)
5353
if avg_loss_value[0] < 10.0:
5454
exit(0) # if avg cost less than 10.0, we think our code is good.
5555
exit(1)

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)