1. fix mp_reader queue, 2. fix erniesagev2 biasattr

2baf006d · suweiyue · 434be194 · 2baf006d
隐藏空白更改
内联并排

Showing with 0 addition and 3 deletion

examples/erniesage/models/erniesage_v2.py examples/erniesage/models/erniesage_v2.py +0 -3

未找到文件。
--- a/examples/erniesage/models/erniesage_v2.py
+++ b/examples/erniesage/models/erniesage_v2.py
@@ -48,9 +48,6 @@ class ErnieSageV2(BaseNet):
            cls = L.fill_constant_batch_size_like(src_feat["term_ids"], [-1, 1, 1], "int64", 1)
            src_ids = L.concat([cls, src_feat["term_ids"]], 1)
            dst_ids = dst_feat["term_ids"]
-            # cls = L.fill_constant_batch_size_like(dst_feat["term_ids"], [-1, 1, 1], "int64", 1)
-            # src_ids = L.concat([cls, dst_feat["term_ids"]], 1)
-            # dst_ids = src_feat["term_ids"]

            # sent_ids
            sent_ids = L.concat([L.zeros_like(src_ids), L.ones_like(dst_ids)], 1)