refine op_test (#12846)

revert-12864-feature/process_lod_grad
chengduo 7 years ago committed by GitHub
parent e8b4e0d627
commit 7748963479
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -56,8 +56,8 @@ def get_numeric_gradient(place,
def get_output():
sum = []
op.run(scope, place)
for output_name in output_names:
op.run(scope, place)
sum.append(
np.array(scope.find_var(output_name).get_tensor()).mean())
return np.array(sum).mean()

@ -153,9 +153,6 @@ def append_input_output(block, op_proto, np_list, is_input, dtype):
def append_loss_ops(block, output_names):
mean_inputs = list(map(block.var, output_names))
# for item in mean_inputs:
# print(item)
# print("Item", item.dtype)
if len(mean_inputs) == 1:
loss = block.create_var(dtype=mean_inputs[0].dtype, shape=[1])

Loading…
Cancel
Save