Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleHub
提交
71defaf0
P
PaddleHub
项目概览
PaddlePaddle
/
PaddleHub
接近 2 年 前同步成功
通知
284
Star
12117
Fork
2091
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
200
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleHub
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
200
Issue
200
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
71defaf0
编写于
7月 22, 2021
作者:
K
KP
提交者:
GitHub
7月 22, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix compatibility issue in machine translation models (#1542)
上级
c052dc67
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
28 addition
and
8 deletion
+28
-8
modules/text/machine_translation/transformer/en-de/README.md
modules/text/machine_translation/transformer/en-de/README.md
+4
-0
modules/text/machine_translation/transformer/en-de/module.py
modules/text/machine_translation/transformer/en-de/module.py
+10
-4
modules/text/machine_translation/transformer/zh-en/README.md
modules/text/machine_translation/transformer/zh-en/README.md
+4
-0
modules/text/machine_translation/transformer/zh-en/module.py
modules/text/machine_translation/transformer/zh-en/module.py
+10
-4
未找到文件。
modules/text/machine_translation/transformer/en-de/README.md
浏览文件 @
71defaf0
...
...
@@ -119,3 +119,7 @@ paddlehub >= 2.1.0
*
1.0.0
初始发布
*
1.0.1
修复模型初始化的兼容性问题
modules/text/machine_translation/transformer/en-de/module.py
浏览文件 @
71defaf0
...
...
@@ -12,13 +12,15 @@
# See the License for the specific language governing permissions and
# limitations under the License.
from
typing
import
List
import
os
from
packaging.version
import
Version
from
typing
import
List
import
paddle
import
paddle.nn
as
nn
from
paddlehub.env
import
MODULE_HOME
from
paddlehub.module.module
import
moduleinfo
,
serving
import
paddlenlp
from
paddlenlp.data
import
Pad
,
Vocab
from
paddlenlp.transformers
import
InferTransformerModel
,
position_encoding_init
...
...
@@ -27,7 +29,7 @@ from transformer_en_de.utils import MTTokenizer, post_process_seq
@
moduleinfo
(
name
=
"transformer_en-de"
,
version
=
"1.0.
0
"
,
version
=
"1.0.
1
"
,
summary
=
""
,
author
=
"PaddlePaddle"
,
author_email
=
""
,
...
...
@@ -42,8 +44,6 @@ class MTTransformer(nn.Layer):
# Model config
model_config
=
{
# Number of sub-layers to be stacked in the encoder and decoder.
"n_layer"
:
6
,
# Number of head used in multi-head attention.
"n_head"
:
8
,
# The dimension for word embeddings, which is also the last dimension of
...
...
@@ -59,6 +59,12 @@ class MTTransformer(nn.Layer):
'dropout'
:
0
}
# Number of sub-layers to be stacked in the encoder and decoder.
if
Version
(
paddlenlp
.
__version__
)
<=
Version
(
'2.0.5'
):
model_config
.
update
({
"n_layer"
:
6
})
else
:
model_config
.
update
({
"num_encoder_layers"
:
6
,
"num_decoder_layers"
:
6
})
# Vocab config
vocab_config
=
{
# Used to pad vocab size to be multiple of pad_factor.
...
...
modules/text/machine_translation/transformer/zh-en/README.md
浏览文件 @
71defaf0
...
...
@@ -117,3 +117,7 @@ paddlehub >= 2.1.0
*
1.0.0
初始发布
*
1.0.1
修复模型初始化的兼容性问题
modules/text/machine_translation/transformer/zh-en/module.py
浏览文件 @
71defaf0
...
...
@@ -12,13 +12,15 @@
# See the License for the specific language governing permissions and
# limitations under the License.
from
typing
import
List
import
os
from
packaging.version
import
Version
from
typing
import
List
import
paddle
import
paddle.nn
as
nn
from
paddlehub.env
import
MODULE_HOME
from
paddlehub.module.module
import
moduleinfo
,
serving
import
paddlenlp
from
paddlenlp.data
import
Pad
,
Vocab
from
paddlenlp.transformers
import
InferTransformerModel
,
position_encoding_init
...
...
@@ -27,7 +29,7 @@ from transformer_zh_en.utils import MTTokenizer, post_process_seq
@
moduleinfo
(
name
=
"transformer_zh-en"
,
version
=
"1.0.
0
"
,
version
=
"1.0.
1
"
,
summary
=
""
,
author
=
"PaddlePaddle"
,
author_email
=
""
,
...
...
@@ -42,8 +44,6 @@ class MTTransformer(nn.Layer):
# Model config
model_config
=
{
# Number of sub-layers to be stacked in the encoder and decoder.
"n_layer"
:
6
,
# Number of head used in multi-head attention.
"n_head"
:
8
,
# The dimension for word embeddings, which is also the last dimension of
...
...
@@ -59,6 +59,12 @@ class MTTransformer(nn.Layer):
'dropout'
:
0
}
# Number of sub-layers to be stacked in the encoder and decoder.
if
Version
(
paddlenlp
.
__version__
)
<=
Version
(
'2.0.5'
):
model_config
.
update
({
"n_layer"
:
6
})
else
:
model_config
.
update
({
"num_encoder_layers"
:
6
,
"num_decoder_layers"
:
6
})
# Vocab config
vocab_config
=
{
# Used to pad vocab size to be multiple of pad_factor.
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录