提交 69859718 编写于 作者: C ceci3

test=develop, change labels name

上级 1b5768c3
...@@ -90,7 +90,7 @@ class TestNpairLossOp(unittest.TestCase): ...@@ -90,7 +90,7 @@ class TestNpairLossOp(unittest.TestCase):
dtype=self.dtype, dtype=self.dtype,
append_batch_size=False) append_batch_size=False)
labels_tensor = fluid.layers.data( labels_tensor = fluid.layers.data(
name='labels', name='labels_t',
shape=[num_data], shape=[num_data],
dtype=self.dtype, dtype=self.dtype,
append_batch_size=False) append_batch_size=False)
...@@ -103,7 +103,7 @@ class TestNpairLossOp(unittest.TestCase): ...@@ -103,7 +103,7 @@ class TestNpairLossOp(unittest.TestCase):
out_tensor = exe.run(feed={ out_tensor = exe.run(feed={
'anchor': embeddings_anchor, 'anchor': embeddings_anchor,
'positive': embeddings_positive, 'positive': embeddings_positive,
'labels': row_labels 'labels_t': row_labels
}, },
fetch_list=[npair_loss_op.name]) fetch_list=[npair_loss_op.name])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册