Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleX
提交
b12b6b4f
P
PaddleX
项目概览
PaddlePaddle
/
PaddleX
通知
138
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
43
列表
看板
标记
里程碑
合并请求
5
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleX
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
43
Issue
43
列表
看板
标记
里程碑
合并请求
5
合并请求
5
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
b12b6b4f
编写于
6月 16, 2020
作者:
S
sunyanfang01
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add prune configs and prompt
上级
07f9ad90
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
2 addition
and
2 deletion
+2
-2
paddlex/cv/models/slim/prune.py
paddlex/cv/models/slim/prune.py
+2
-2
未找到文件。
paddlex/cv/models/slim/prune.py
浏览文件 @
b12b6b4f
...
...
@@ -158,7 +158,7 @@ def prune_program(model, prune_params_ratios=None):
prune_params_ratios (dict): 由裁剪参数名和裁剪率组成的字典,当为None时
使用默认裁剪参数名和裁剪率。默认为None。
"""
assert
model
.
status
==
'Normal'
,
'Only the model
after training can be prun
ed!'
assert
model
.
status
==
'Normal'
,
'Only the model
s saved while training are support
ed!'
place
=
model
.
places
[
0
]
train_prog
=
model
.
train_prog
eval_prog
=
model
.
test_prog
...
...
@@ -236,7 +236,7 @@ def cal_params_sensitivities(model, save_file, eval_dataset, batch_size=8):
其中``weight_0``是卷积Kernel名;``sensitivities['weight_0']``是一个字典,key是裁剪率,value是敏感度。
"""
assert
model
.
status
==
'Normal'
,
'Only the model
after training can calculate sensitivities data
!'
assert
model
.
status
==
'Normal'
,
'Only the model
s saved while training are supported
!'
if
os
.
path
.
exists
(
save_file
):
os
.
remove
(
save_file
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录