神器!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”获取


个人微信
请注明:地区+学校/企业+研究方向+昵称
如果没有备注不拉群!

浏览 13
点赞
评论
收藏
分享

手机扫一扫分享

举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

举报