Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
139554da
X
X2Paddle
项目概览
PaddlePaddle
/
X2Paddle
1 年多 前同步成功
通知
328
Star
698
Fork
167
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
26
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
26
Issue
26
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
139554da
编写于
11月 30, 2021
作者:
W
wjj19950828
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
deal with comments
上级
573b7b14
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
6 addition
and
6 deletion
+6
-6
x2paddle/convert.py
x2paddle/convert.py
+3
-3
x2paddle/core/program.py
x2paddle/core/program.py
+3
-3
未找到文件。
x2paddle/convert.py
浏览文件 @
139554da
...
...
@@ -88,7 +88,7 @@ def arg_parser():
default
=
None
,
help
=
"pretrain model file of pytorch model"
)
parser
.
add_argument
(
"--
code_optimizer
"
,
"--
enable_code_optim
"
,
"-co"
,
default
=
True
,
help
=
"Turn on code optimization"
)
...
...
@@ -225,7 +225,7 @@ def pytorch2paddle(module,
save_dir
,
jit_type
=
"trace"
,
input_examples
=
None
,
code_optimizer
=
True
,
enable_code_optim
=
True
,
convert_to_lite
=
False
,
lite_valid_places
=
"arm"
,
lite_model_type
=
"naive_buffer"
):
...
...
@@ -260,7 +260,7 @@ def pytorch2paddle(module,
graph_opt
.
optimize
(
mapper
.
paddle_graph
)
print
(
"Model optimized."
)
mapper
.
paddle_graph
.
gen_model
(
save_dir
,
jit_type
=
jit_type
,
code_optimizer
=
code_optimizer
)
save_dir
,
jit_type
=
jit_type
,
enable_code_optim
=
enable_code_optim
)
if
convert_to_lite
:
convert2lite
(
save_dir
,
lite_valid_places
,
lite_model_type
)
...
...
x2paddle/core/program.py
浏览文件 @
139554da
...
...
@@ -237,11 +237,11 @@ class PaddleGraph(object):
return
update
(
self
.
layers
)
def
gen_model
(
self
,
save_dir
,
jit_type
=
None
,
code_optimizer
=
True
):
def
gen_model
(
self
,
save_dir
,
jit_type
=
None
,
enable_code_optim
=
True
):
if
not
osp
.
exists
(
save_dir
):
os
.
makedirs
(
save_dir
)
if
jit_type
==
"trace"
:
if
not
self
.
has_unpack
and
code_optimizer
:
if
not
self
.
has_unpack
and
enable_code_optim
:
from
x2paddle.optimizer.pytorch_code_optimizer
import
HierarchicalTree
hierarchical_tree
=
HierarchicalTree
(
self
)
for
layer_id
,
layer
in
self
.
layers
.
items
():
...
...
@@ -252,7 +252,7 @@ class PaddleGraph(object):
self
.
gen_code
(
save_dir
)
self
.
dump_parameter
(
save_dir
)
else
:
if
self
.
source_type
==
"pytorch"
and
code_optimizer
:
if
self
.
source_type
==
"pytorch"
and
enable_code_optim
:
from
x2paddle.optimizer.pytorch_code_optimizer
import
ModuleGraph
module_graph
=
ModuleGraph
(
self
)
module_graph
.
save_source_files
(
save_dir
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录