network-slimming 弱弱的问下，ResNet164这个模型哪里下的。。

弱弱的问下，ResNet164这个模型哪里下的。。

Open fsj0122 opened this issue 4 years ago • 12 comments

Feb 08 '21 02:02 fsj0122

你的意思是模型定义从哪来的? https://github.com/bearpaw/pytorch-classification/blob/master/models/cifar/preresnet.py

Feb 08 '21 16:02 Eric-mingjie

第三步，python resprune.py --dataset cifar10 --depth 164 --percent 0.4 --model [PATH TO THE MODEL]，这里PATH TO THE MODEL得MODEL是从什么地方得到的？

Feb 09 '21 03:02 fsj0122

你的意思是模型定义从哪来的? https://github.com/bearpaw/pytorch-classification/blob/master/models/cifar/preresnet.py

是的，第三步PRUNE这里，PATH TO THE MODEL，我想知道ResNet164这个模型文件是怎么得到的。

Feb 09 '21 03:02 fsj0122

上面train with sparsity会存下模型

Feb 09 '21 04:02 liuzhuang13

上面train with sparsity会存下模型

请问一下，这个基于cifar10的数据集可以改成coco数据集么？

Feb 09 '21 12:02 fsj0122

应该可以，不过参数什么的要另外调

Feb 10 '21 17:02 Eric-mingjie

应该可以，不过参数什么的要另外调

但是dataset根本没有coco数据集的选项啊，是不是要在main.py里加上coco数据集的if判断？

Feb 22 '21 02:02 fsj0122

这个coco的代码要自己写，我们的repo没有支持coco。

Feb 23 '21 18:02 Eric-mingjie

这个coco的代码要自己写，我们的repo没有支持coco。

在你们的代码基础上，加上coco的部分么？

Feb 26 '21 08:02 fsj0122

这个应该也可以吧，也可以单独写，看你觉得怎么方便

Feb 26 '21 18:02 Eric-mingjie

这个应该也可以吧，也可以单独写，看你觉得怎么方便

代码里有assert（depth-2）%9 == 0，导致无法生成resnet50模型，怎么办？

Mar 05 '21 14:03 fsj0122

稍微改一下就行，resnet50是用basicblock的

Mar 09 '21 18:03 Eric-mingjie