search

Eric-mingjie / network-slimming

Network Slimming (Pytorch) (ICCV 2017)
MIT License
907 stars 214 forks source link

原论文中的memory是指 #81

Open PQ-girl opened 2 years ago

PQ-girl commented 2 years ago

原论文中的表6中的memory指的是什么？我训练出的vgg16的权重文件只有118MB,可表6中的baseline的memory为697MB，这两个不是一个东西吗

Eric-mingjie commented 2 years ago

表6是figure 6吗, figure 6 里没有说memory. https://openaccess.thecvf.com/content_ICCV_2017/papers/Liu_Learning_Efficient_Convolutional_ICCV_2017_paper.pdf

PQ-girl commented 2 years ago

我下载的论文里有附录，附录里面有个表6 捕获

PQ-girl commented 2 years ago

你好，为什么我把图片大小改为224*224后，剪枝时会报错，不过可以运行一半，文件pruned都能保存，但是最后还是报错，说维度不匹配，可以帮帮我吗

Eric-mingjie commented 2 years ago

memory不与原文的数字吻合可能是因为不同的ml框架，原论文的数字是基于torch, 而本论文是pytorch实现。

Eric-mingjie commented 2 years ago

这个repo没有imagenet的network slimming实现，可参考https://github.com/Eric-mingjie/rethinking-network-pruning/tree/master/imagenet/network-slimming。

PQ-girl commented 2 years ago

你好，我试了ImageNet里面的network—sliming的方法，微调后为什么又返回原来的尺寸了，还有ImageNet里面的vgg比这里的vgg网络大的多，但是为什么它训练出来的准确率反而低了呢，并且它的输入尺寸还是224×224的。还有能告诉我这里的prune代码和ImageNet里面224×224的prune哪里不同，我想换成224×224的，但一直报错，您可以帮帮我吗