Commit b8e9759b by libei

fix convert model bugs when convert dense_relative_transformer models

parent 17822517
...@@ -473,7 +473,7 @@ def convert_settings(settings): ...@@ -473,7 +473,7 @@ def convert_settings(settings):
args['arch'] = 'relative_transformer' args['arch'] = 'relative_transformer'
if use_relative_position_representation and use_dense: if use_relative_position_representation and use_dense:
args['arch'] = 'relative_dense_transformer' args['arch'] = 'dense_relative_transformer'
return argparse.Namespace(**args) return argparse.Namespace(**args)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论