栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 系统运维 > 运维 > Linux

Ubuntu下PocketFlow模型压缩实践

Linux 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Ubuntu下PocketFlow模型压缩实践

安装PocketFlow

博主使用的是Python3.6,TensorFlow1.12

git clone https://github.com/Tencent/PocketFlow.git
cd PocketFlow
数据集准备

进行datasets文件夹路径可以看到,PocketFlow支持的数据集有cifar10下载链接,ILSVRC-12和PascalVoc官方链接。其中cifar10是图片分类任务,ILSVRC-12用于分类任务和目标检测任务,PascalVoc是深度学习目标检测的数据集。

这里以cifar10数据为例测试。

tar zxvf cifar-10-binary.tar.gz
修改配置文件参数
cp path.conf.template path.conf 

然后修改path.conf 文件中的cifar数据集路径,(运行在local版本),更新后如下:

# data files
data_hdfs_host = None
data_dir_local_cifar10 = /XXX/Dataset/cifar-10-batches-bin
data_dir_hdfs_cifar10 = None
data_dir_seven_cifar10 = None
data_dir_docker_cifar10 = /opt/ml/data  # DO NOT EDIT
data_dir_local_ilsvrc12 = /mnt/inspurfs/user-fs/wenjing/Dataset/ILSVRC
data_dir_hdfs_ilsvrc12 = Nonea
data_dir_seven_ilsvrc12 = Noneaa
data_dir_docker_ilsvrc12 = /opt/ml/data  # DO NOT EDIT

# model filesa
model_http_url = https://api.ai.tencent.com/pocketflow
全模型训练

由于官方提供的预训练模型链接已经不在支持,需要在本地去训练一个完整的网络模型。

./scripts/run_local.sh nets/resnet_at_cifar10_run.py  
--resnet_size=20

训练好的模型可以在./models路径中找到

模型压缩(剪枝)
./scripts/run_local.sh nets/resnet_at_cifar10_run.py  
--learner=chn-pruned-gpu 
--cp_uniform_preserve_ratio=0.5 
--cp_prune_option=uniform 
--resnet_size=20
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/893728.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号