未验证 提交 c701d2b8 编写于 作者: B Bai Yifan 提交者: GitHub

fix test_fsp_loss ut (#668)

Co-authored-by: Nceci3 <ceci3@users.noreply.github.com>
上级 fe145d73
......@@ -44,13 +44,14 @@ class TestFSPLoss(StaticCase):
data_name_map = {'image': 'image'}
merge(teacher_main,
paddle.static.default_main_program(), data_name_map, place)
merged_ops = []
for block in paddle.static.default_main_program().blocks:
for op in block.ops:
merged_ops.append(op.type)
distill_loss = fsp_loss(
'teacher_conv5_bn_output.tmp_2', 'teacher_conv6_bn_output.tmp_2',
'conv1_bn_output.tmp_2', 'conv2_bn_output.tmp_2')
distill_loss = fsp_loss('teacher_conv1_out.tmp_1',
'teacher_conv6_out.tmp_0', 'conv1_out.tmp_0',
'conv2_out.tmp_0')
loss_ops = []
for block in paddle.static.default_main_program().blocks:
for op in block.ops:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册