helinwang-patch-1
Yu Yang 7 years ago
parent e18a269705
commit 389ea18a4e

@ -71,12 +71,13 @@ class ParallelExecutor(unittest.TestCase):
first_loss = numpy.array(fluid.global_scope().find_var('fetched_var') first_loss = numpy.array(fluid.global_scope().find_var('fetched_var')
.get_lod_tensor_array()[0]) .get_lod_tensor_array()[0])
print first_loss
for i in xrange(10): #
exe.run([], 'fetched_var') # for i in xrange(10):
exe.run([loss.name], 'fetched_var') # exe.run([], 'fetched_var')
last_loss = numpy.array(fluid.global_scope().find_var('fetched_var') # exe.run([loss.name], 'fetched_var')
.get_lod_tensor_array()[0]) # last_loss = numpy.array(fluid.global_scope().find_var('fetched_var')
# .get_lod_tensor_array()[0])
print first_loss, last_loss #
self.assertGreater(first_loss[0], last_loss[0]) # print first_loss, last_loss
# self.assertGreater(first_loss[0], last_loss[0])

Loading…
Cancel
Save