Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PGL
提交
434be194
P
PGL
项目概览
PaddlePaddle
/
PGL
通知
76
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
11
列表
看板
标记
里程碑
合并请求
1
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PGL
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
11
Issue
11
列表
看板
标记
里程碑
合并请求
1
合并请求
1
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
434be194
编写于
5月 27, 2020
作者:
S
suweiyue
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
1. fix mp_reader queue, 2. fix erniesagev2 biasattr
上级
5782fb81
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
25 addition
and
12 deletion
+25
-12
examples/erniesage/models/erniesage_v2.py
examples/erniesage/models/erniesage_v2.py
+8
-3
examples/erniesage/models/message_passing.py
examples/erniesage/models/message_passing.py
+16
-8
pgl/utils/mp_reader.py
pgl/utils/mp_reader.py
+1
-1
未找到文件。
examples/erniesage/models/erniesage_v2.py
浏览文件 @
434be194
...
...
@@ -48,6 +48,9 @@ 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
)
...
...
@@ -81,14 +84,16 @@ class ErnieSageV2(BaseNet):
self_feature
=
L
.
fc
(
self_feature
,
hidden_size
,
act
=
act
,
param_attr
=
F
.
ParamAttr
(
name
=
name
+
"_l"
,
param_attr
=
F
.
ParamAttr
(
name
=
name
+
"_l
.w_0
"
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_l.b_0"
)
neigh_feature
=
L
.
fc
(
neigh_feature
,
hidden_size
,
act
=
act
,
param_attr
=
F
.
ParamAttr
(
name
=
name
+
"_r"
,
learning_rate
=
learning_rate
),
param_attr
=
F
.
ParamAttr
(
name
=
name
+
"_r.w_0"
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_r.b_0"
)
output
=
L
.
concat
([
self_feature
,
neigh_feature
],
axis
=
1
)
output
=
L
.
l2_normalize
(
output
,
axis
=
1
)
...
...
examples/erniesage/models/message_passing.py
浏览文件 @
434be194
...
...
@@ -57,14 +57,16 @@ def graphsage_sum(gw, feature, hidden_size, act, initializer, learning_rate, nam
self_feature
=
fluid
.
layers
.
fc
(
self_feature
,
hidden_size
,
act
=
act
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_l"
,
initializer
=
initializer
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_l
.w_0
"
,
initializer
=
initializer
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_l.b_0"
)
neigh_feature
=
fluid
.
layers
.
fc
(
neigh_feature
,
hidden_size
,
act
=
act
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_r"
,
initializer
=
initializer
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_r
.w_0
"
,
initializer
=
initializer
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_r.b_0"
)
output
=
fluid
.
layers
.
concat
([
self_feature
,
neigh_feature
],
axis
=
1
)
output
=
fluid
.
layers
.
l2_normalize
(
output
,
axis
=
1
)
...
...
@@ -79,14 +81,16 @@ def graphsage_mean(gw, feature, hidden_size, act, initializer, learning_rate, na
self_feature
=
fluid
.
layers
.
fc
(
self_feature
,
hidden_size
,
act
=
act
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_l"
,
initializer
=
initializer
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_l
.w_0
"
,
initializer
=
initializer
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_l.b_0"
)
neigh_feature
=
fluid
.
layers
.
fc
(
neigh_feature
,
hidden_size
,
act
=
act
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_r"
,
initializer
=
initializer
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_r
.w_0
"
,
initializer
=
initializer
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_r.b_0"
)
output
=
fluid
.
layers
.
concat
([
self_feature
,
neigh_feature
],
axis
=
1
)
output
=
fluid
.
layers
.
l2_normalize
(
output
,
axis
=
1
)
...
...
@@ -101,14 +105,16 @@ def pinsage_mean(gw, feature, hidden_size, act, initializer, learning_rate, name
self_feature
=
fluid
.
layers
.
fc
(
self_feature
,
hidden_size
,
act
=
act
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_l"
,
initializer
=
initializer
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_l
.w_0
"
,
initializer
=
initializer
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_l.b_0"
)
neigh_feature
=
fluid
.
layers
.
fc
(
neigh_feature
,
hidden_size
,
act
=
act
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_r"
,
initializer
=
initializer
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_r
.w_0
"
,
initializer
=
initializer
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_r.b_0"
)
output
=
fluid
.
layers
.
concat
([
self_feature
,
neigh_feature
],
axis
=
1
)
output
=
fluid
.
layers
.
l2_normalize
(
output
,
axis
=
1
)
...
...
@@ -123,14 +129,16 @@ def pinsage_sum(gw, feature, hidden_size, act, initializer, learning_rate, name)
self_feature
=
fluid
.
layers
.
fc
(
self_feature
,
hidden_size
,
act
=
act
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_l"
,
initializer
=
initializer
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_l
.w_0
"
,
initializer
=
initializer
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_l.b_0"
)
neigh_feature
=
fluid
.
layers
.
fc
(
neigh_feature
,
hidden_size
,
act
=
act
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_r"
,
initializer
=
initializer
,
param_attr
=
fluid
.
ParamAttr
(
name
=
name
+
"_r
.w_0
"
,
initializer
=
initializer
,
learning_rate
=
learning_rate
),
bias_attr
=
name
+
"_r.b_0"
)
output
=
fluid
.
layers
.
concat
([
self_feature
,
neigh_feature
],
axis
=
1
)
output
=
fluid
.
layers
.
l2_normalize
(
output
,
axis
=
1
)
...
...
pgl/utils/mp_reader.py
浏览文件 @
434be194
...
...
@@ -25,7 +25,7 @@ except:
import
numpy
as
np
import
time
import
paddle.fluid
as
fluid
from
queue
import
Queue
from
multiprocessing
import
Queue
import
threading
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录