-
How to do the pruning step for batchNorm1D layers in an ANN, where you would be using the weights directly rather than using a mask?
If possible, a sample code on the nn.Linear layer and the BatchNor…
-
### Checklist
- I have searched related issues but cannot get the expected help.
- I have read related documents and don't know what to do.
### Describe the question you meet
I used the offi…
-
Hello, would you please give me a hand? I met a tough question when I tried to use the method to prune the channels of VGG16_reducedfc, the basebone of SSD. I was stuck in the problem for several week…
-
Hello, would you please give me a hand? I met a tough question when I tried to use a [method](https://github.com/Eric-mingjie/network-slimming) to prune the channels of VGG16_reducedfc, the basebone o…
-
-
这代码可以正常训练?为什么我的报错误呢?
Traceback (most recent call last):
File "/media/data/liuben/optimize/yolov3-network-slimming/sparsity_train.py", line 164, in
train()
File "/media/data/liuben/optimiz…
-
Hi:
我在训练的时候到这里报错:
loss_cls = (1 / nB) * self.ce_loss(pred_cls[mask], torch.argmax(tcls[mask], dim=1))
Traceback (most recent call last):
File "/home/lc/work/yolov3-network-slimming/sparsit…
-
https://github.com/Eric-mingjie/network-slimming/blob/98e6b4d28e9909eff22ce4c45cba1b605cac9d65/vggprune.py#L47
erhsh updated
3 years ago
-
after network-slimming,the size of modell are the same as premodel?
-
Hello, when I used the FLOPs calculation method in the link [https://github.com/Eric-mingjie/rethinking-network-pruning/blob/master/imagenet/l1-norm-pruning/compute_flops.py](url) you sent to calculat…