Comments (1)
kame-lqm
commented on July 30, 2024
1
已经搞定了，主要还是要在train.py里面修改如下一行代码就OK了。测试发现稀疏训练跟剪枝都有效果。0.5剪枝基本没有掉点，甚至还涨了一丢丢。
model = torch.nn.parallel.DistributedDataParallel(model)
改为
model = torch.nn.parallel.DistributedDataParallel(model, find_unused_parameters=True)
from yolov3-channel-and-layer-pruning.
Related Issues (20)
- Can I use this to prune model MobileNetv2-Yolov4
- ValueError: not enough values to unpack (expected 2, got 1)
- 关于稀疏训练和剪枝的阈值
- 进行yolov4稀疏化训练 HOT 3
- 关于其他数据集标注格式要进行的代码修改
- 为什么稀疏训练和剪枝都要将yolov3赋给darknet？
- What are the flags -sr ,--s 0.001 ,--prune 1 --global_percent flag while doing pruning
- 刚开始推理时，GPU内存占用持续增加
- 训练时候出现一个问题 soft rratio是0测试时候 P R mAP F1也是nan这种情况怎么办呢
- The P R F1-score and [email protected] are around 90% after sparse training (--s 0.001 --prune 0). Still the sparse model is not performing well on the test dataset.
- 通道剪枝的时候遇到问题~ HOT 1
- 请问支持分组卷积的枝剪吗 HOT 1
- data文件夹下的.shapes文件需要自己准备吗？
- ValueError: Number of rows must be a positive integer, not 4.0 HOT 3
- 剪枝改进后的模型
- RuntimeError: shape '[128, 64, 3, 3]' is invalid for input of size 7067 训练v3报错
- 可以正常剪枝，但是微调并没有提升map，且微调后的bn_weight的大部分值仍接近0？
- How to transfor the pt model to ONNX format
- 在windows下没有生成val.shapes
- 没有models.yolo
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 yolov3-channel-and-layer-pruning.