search

Eric-mingjie / network-slimming

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

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

Open fsj0122 opened 3 years ago

Eric-mingjie commented 3 years ago

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

fsj0122 commented 3 years ago

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

fsj0122 commented 3 years ago

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

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

liuzhuang13 commented 3 years ago

上面train with sparsity会存下模型

fsj0122 commented 3 years ago

上面train with sparsity会存下模型

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

Eric-mingjie commented 3 years ago

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

fsj0122 commented 3 years ago

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

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

Eric-mingjie commented 3 years ago

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

fsj0122 commented 3 years ago

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

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

Eric-mingjie commented 3 years ago

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

fsj0122 commented 3 years ago

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

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

Eric-mingjie commented 3 years ago

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