Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PALM
提交
42f9ba10
P
PALM
项目概览
PaddlePaddle
/
PALM
通知
5
Star
3
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
10
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PALM
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
10
Issue
10
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
42f9ba10
编写于
10月 30, 2019
作者:
X
xixiaoyao
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix bugs
上级
ada30bc3
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
4 addition
and
4 deletion
+4
-4
paddlepalm/task_paradigm/mlm.py
paddlepalm/task_paradigm/mlm.py
+1
-0
paddlepalm/task_paradigm/mrc.py
paddlepalm/task_paradigm/mrc.py
+2
-0
run_demo2.sh
run_demo2.sh
+1
-4
未找到文件。
paddlepalm/task_paradigm/mlm.py
浏览文件 @
42f9ba10
...
@@ -57,6 +57,7 @@ class TaskParadigm(task_paradigm):
...
@@ -57,6 +57,7 @@ class TaskParadigm(task_paradigm):
# 多任务学习时才需要引入这个,防止其他run其他任务时导致seqlen过小,gather超范围
# 多任务学习时才需要引入这个,防止其他run其他任务时导致seqlen过小,gather超范围
max_position
=
inputs
[
"reader"
][
"batchsize_x_seqlen"
]
-
1
max_position
=
inputs
[
"reader"
][
"batchsize_x_seqlen"
]
-
1
mask_pos
=
fluid
.
layers
.
elementwise_min
(
mask_pos
,
max_position
)
mask_pos
=
fluid
.
layers
.
elementwise_min
(
mask_pos
,
max_position
)
mask_pos
.
stop_gradient
=
True
word_emb
=
inputs
[
"backbone"
][
"embedding_table"
]
word_emb
=
inputs
[
"backbone"
][
"embedding_table"
]
enc_out
=
inputs
[
"backbone"
][
"encoder_outputs"
]
enc_out
=
inputs
[
"backbone"
][
"encoder_outputs"
]
...
...
paddlepalm/task_paradigm/mrc.py
浏览文件 @
42f9ba10
...
@@ -80,6 +80,8 @@ class TaskParadigm(task_paradigm):
...
@@ -80,6 +80,8 @@ class TaskParadigm(task_paradigm):
max_position
=
inputs
[
"reader"
][
"seqlen"
]
-
1
max_position
=
inputs
[
"reader"
][
"seqlen"
]
-
1
start_positions
=
fluid
.
layers
.
elementwise_min
(
start_positions
,
max_position
)
start_positions
=
fluid
.
layers
.
elementwise_min
(
start_positions
,
max_position
)
end_positions
=
fluid
.
layers
.
elementwise_min
(
end_positions
,
max_position
)
end_positions
=
fluid
.
layers
.
elementwise_min
(
end_positions
,
max_position
)
start_positions
.
stop_gradient
=
True
end_positions
.
stop_gradient
=
True
else
:
else
:
unique_id
=
inputs
[
'reader'
][
'unique_ids'
]
unique_id
=
inputs
[
'reader'
][
'unique_ids'
]
...
...
run_demo2.sh
浏览文件 @
42f9ba10
export
CUDA_VISIBLE_DEVICES
=
0,1,2,3,4,5,6,7
export
CUDA_VISIBLE_DEVICES
=
0,1,2,3,4,5,6,7
while
true
python
-u
demo2.py
do
python
-u
demo2.py
done
# GLOG_vmodule=lookup_table_op=4 python -u demo2.py > debug2.log 2>&1
# GLOG_vmodule=lookup_table_op=4 python -u demo2.py > debug2.log 2>&1
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录