神器!Pytorch结构化神经网络修剪工具包
AI算法与图像处理
共 467字,需浏览 1分钟
·
2020-08-01 05:19
点击上方“AI算法与图像处理”,选择加"星标"或“置顶”
重磅干货,第一时间送达
来源 | 程序员遇见GitHub
作者 | 阿三
大家好,今天给大家带来的是Pytorch结构化神经网络修建工具包。
一. Torch-Pruning
这是一个在pytorch中进行结构化神经网络修剪和layer dependency的维护的工具包,能够处理各种网络,如DenseNet、ResNet和Inception。
GitHub地址:
https://github.com/VainF/Torch-Pruning
最后的最后求一波分享!
YOLOv4 trick相关论文已经下载并放在公众号后台
关注“AI算法与图像处理”,回复 “200714”获取
个人微信 请注明:地区+学校/企业+研究方向+昵称 如果没有备注不拉群!
评论