Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSlim
提交
fd5084c6
P
PaddleSlim
项目概览
PaddlePaddle
/
PaddleSlim
1 年多 前同步成功
通知
51
Star
1434
Fork
344
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
16
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSlim
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
16
合并请求
16
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
fd5084c6
编写于
8月 09, 2021
作者:
W
whs
提交者:
GitHub
8月 09, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix ce of dygraph quant (#871)
上级
e3663d26
变更
6
显示空白变更内容
内联
并排
Showing
6 changed file
with
7 addition
and
7 deletion
+7
-7
ce_tests/dygraph/quant/readme.md
ce_tests/dygraph/quant/readme.md
+1
-1
ce_tests/dygraph/quant/run_ptq.sh
ce_tests/dygraph/quant/run_ptq.sh
+1
-1
ce_tests/dygraph/quant/run_qat.sh
ce_tests/dygraph/quant/run_qat.sh
+2
-2
ce_tests/dygraph/quant/src/ptq.py
ce_tests/dygraph/quant/src/ptq.py
+1
-1
ce_tests/dygraph/quant/src/qat.py
ce_tests/dygraph/quant/src/qat.py
+1
-1
ce_tests/dygraph/quant/src/test.py
ce_tests/dygraph/quant/src/test.py
+1
-1
未找到文件。
ce_tests/dygraph/quant/readme.md
浏览文件 @
fd5084c6
...
@@ -2,7 +2,7 @@
...
@@ -2,7 +2,7 @@
安装需要测试的Paddle版本和PaddleSlim版本。
安装需要测试的Paddle版本和PaddleSlim版本。
准备ImageNet数据集,数据集需要满足paddle hapi的要求。假定解压到
`/dataset/ILSVRC2012`
文件夹,该文件夹下有
`train文件夹、val
_hapi文件夹、train_list.txt和val_list.txt文件`
。如果数据集有问题,可以私聊
。
准备ImageNet数据集,数据集需要满足paddle hapi的要求。假定解压到
`/dataset/ILSVRC2012`
文件夹,该文件夹下有
`train文件夹、val
文件夹、train_list.txt和val_list.txt文件`
。
通过
`export CUDA_VISIBLE_DEVICES=xx`
指定需要使用的GPU ID。
通过
`export CUDA_VISIBLE_DEVICES=xx`
指定需要使用的GPU ID。
...
...
ce_tests/dygraph/quant/run_ptq.sh
浏览文件 @
fd5084c6
data_path
=
"/
dataset
/ILSVRC2012"
data_path
=
"/
root/datasets
/ILSVRC2012"
quant_batch_num
=
10
quant_batch_num
=
10
quant_batch_size
=
10
quant_batch_size
=
10
...
...
ce_tests/dygraph/quant/run_qat.sh
浏览文件 @
fd5084c6
data_path
=
"/
dataset
/ILSVRC2012"
data_path
=
"/
root/datasets
/ILSVRC2012"
val_dir
=
"val
_hapi
"
val_dir
=
"val"
epoch
=
1
epoch
=
1
lr
=
0.0001
lr
=
0.0001
batch_size
=
32
batch_size
=
32
...
...
ce_tests/dygraph/quant/src/ptq.py
浏览文件 @
fd5084c6
...
@@ -100,7 +100,7 @@ if __name__ == '__main__':
...
@@ -100,7 +100,7 @@ if __name__ == '__main__':
)
)
parser
.
add_argument
(
parser
.
add_argument
(
'--val_dir'
,
'--val_dir'
,
default
=
"val
_hapi
"
,
default
=
"val"
,
help
=
'the dir that saves val images for paddle.Model'
)
help
=
'the dir that saves val images for paddle.Model'
)
# train
# train
...
...
ce_tests/dygraph/quant/src/qat.py
浏览文件 @
fd5084c6
...
@@ -154,7 +154,7 @@ if __name__ == '__main__':
...
@@ -154,7 +154,7 @@ if __name__ == '__main__':
'(should have subdirectories named "train" and "val"'
)
'(should have subdirectories named "train" and "val"'
)
parser
.
add_argument
(
parser
.
add_argument
(
'--val_dir'
,
'--val_dir'
,
default
=
"val
_hapi
"
,
default
=
"val"
,
help
=
'the dir that saves val images for paddle.Model'
)
help
=
'the dir that saves val images for paddle.Model'
)
# train
# train
...
...
ce_tests/dygraph/quant/src/test.py
浏览文件 @
fd5084c6
...
@@ -35,7 +35,7 @@ def eval(args):
...
@@ -35,7 +35,7 @@ def eval(args):
# prepare data
# prepare data
val_dataset
=
dataset
.
ImageNetDataset
(
val_dataset
=
dataset
.
ImageNetDataset
(
path
=
os
.
path
.
join
(
args
.
data_dir
,
'val
_hapi
'
),
mode
=
'val'
)
path
=
os
.
path
.
join
(
args
.
data_dir
,
'val'
),
mode
=
'val'
)
eval_loader
=
paddle
.
io
.
DataLoader
(
eval_loader
=
paddle
.
io
.
DataLoader
(
val_dataset
,
batch_size
=
args
.
batch_size
,
num_workers
=
5
)
val_dataset
,
batch_size
=
args
.
batch_size
,
num_workers
=
5
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录