Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
4091aca7
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
1 年多 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
4091aca7
编写于
10月 19, 2021
作者:
W
Walter
提交者:
GitHub
10月 19, 2021
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #1314 from RainFrost1/release/2.3
[cherry pick]fix slim load pretrained model bug
上级
38bdf3c8
14c43838
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
5 addition
and
4 deletion
+5
-4
ppcls/engine/engine.py
ppcls/engine/engine.py
+5
-4
未找到文件。
ppcls/engine/engine.py
浏览文件 @
4091aca7
...
@@ -183,6 +183,11 @@ class Engine(object):
...
@@ -183,6 +183,11 @@ class Engine(object):
self
.
model
=
build_model
(
self
.
config
[
"Arch"
])
self
.
model
=
build_model
(
self
.
config
[
"Arch"
])
# set @to_static for benchmark, skip this by default.
# set @to_static for benchmark, skip this by default.
apply_to_static
(
self
.
config
,
self
.
model
)
apply_to_static
(
self
.
config
,
self
.
model
)
# for slim
self
.
pruner
=
get_pruner
(
self
.
config
,
self
.
model
)
self
.
quanter
=
get_quaner
(
self
.
config
,
self
.
model
)
# load_pretrain
# load_pretrain
if
self
.
config
[
"Global"
][
"pretrained_model"
]
is
not
None
:
if
self
.
config
[
"Global"
][
"pretrained_model"
]
is
not
None
:
if
self
.
config
[
"Global"
][
"pretrained_model"
].
startswith
(
"http"
):
if
self
.
config
[
"Global"
][
"pretrained_model"
].
startswith
(
"http"
):
...
@@ -192,10 +197,6 @@ class Engine(object):
...
@@ -192,10 +197,6 @@ class Engine(object):
load_dygraph_pretrain
(
load_dygraph_pretrain
(
self
.
model
,
self
.
config
[
"Global"
][
"pretrained_model"
])
self
.
model
,
self
.
config
[
"Global"
][
"pretrained_model"
])
# for slim
self
.
pruner
=
get_pruner
(
self
.
config
,
self
.
model
)
self
.
quanter
=
get_quaner
(
self
.
config
,
self
.
model
)
# build optimizer
# build optimizer
if
self
.
mode
==
'train'
:
if
self
.
mode
==
'train'
:
self
.
optimizer
,
self
.
lr_sch
=
build_optimizer
(
self
.
optimizer
,
self
.
lr_sch
=
build_optimizer
(
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录