Comments (1)
估计是裁剪完的网络通道数不是2^n,导致再GPU上并行加速效果下降,推理时间就增加了。CPU上应该会有速度提升。
Originally posted by @Yaodada12 in #89 (comment) 您好,我测试了您说明的情况发现CPU上确实是更快的,但是还是想了解,在这种情况下有什么办法可以加快GPU上的推理速度吗?还是说只有调大剪枝率才可以呢?
尝试控制剪枝后的通道数为2^n,速度不满意根据场景看能不能用TensorRT加速。
from yolov5prune.
Related Issues (20)
- 我想请教下稀疏训练的稀疏效果一直不好是怎么回事,我甚至把sr开到了10 HOT 2
- prune.py脚本运行报错 too many indices for tensor demension 3 HOT 1
- 计算量 HOT 1
- RuntimeError: mat1 and mat2 shapes cannot be multiplied HOT 1
- 训练自己的算法
- 使用train_sparsity.py出现的错误- HOT 1
- 已经充分稀疏的情况下,剪枝率只能设置到0.1
- 关于train.py训练后tensorboard中没有bn_weights/hist直方图 HOT 1
- v7.0版本能用吗? HOT 4
- 运行预测脚本的时候报错了 显示yolo脚本没有DetectionModel这个属性
- 更换主干网络再经行稀疏化训练后,值变成了nan
- tensorboard中的histogram图表中的数据如何导出呢
- IndexError: too many indices for tensor of dimension 3
- 稀疏因子最高只能设为0.1,再高一点剪枝最大比例反而会降低是怎么回事
- 剪枝出现Scanning 'E:\task\experiment\dataset\BITVehicle\BITVehicle\val.cache' images and labels... 0 found, 985 missing, 0 empty, 0 corrupted HOT 1
- 博主麻烦看看这个问题 进行prune.py 出现KeyError: 'model.0.bn'问题
- Can't get attribute 'ModelPruned' on <module 'models.yolo'
- gamma, bias distribution
- 剪枝完之后在GPU上的推理速度反倒比剪枝前的模型更慢
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from yolov5prune.