Open fsj0122 opened 3 years ago
第三步,python resprune.py --dataset cifar10 --depth 164 --percent 0.4 --model [PATH TO THE MODEL],这里PATH TO THE MODEL得MODEL是从什么地方得到的?
你的意思是模型定义从哪来的? https://github.com/bearpaw/pytorch-classification/blob/master/models/cifar/preresnet.py
是的,第三步PRUNE这里,PATH TO THE MODEL,我想知道ResNet164这个模型文件是怎么得到的。
上面train with sparsity会存下模型
上面train with sparsity会存下模型
请问一下,这个基于cifar10的数据集可以改成coco数据集么?
应该可以,不过参数什么的要另外调
应该可以,不过参数什么的要另外调
但是dataset根本没有coco数据集的选项啊,是不是要在main.py里加上coco数据集的if判断?
这个coco的代码要自己写, 我们的repo没有支持coco。
这个coco的代码要自己写, 我们的repo没有支持coco。
在你们的代码基础上,加上coco的部分么?
这个应该也可以吧,也可以单独写,看你觉得怎么方便
这个应该也可以吧,也可以单独写,看你觉得怎么方便
代码里有assert(depth-2)%9 == 0,导致无法生成resnet50模型,怎么办?
稍微改一下就行,resnet50是用basicblock的
你的意思是模型定义从哪来的? https://github.com/bearpaw/pytorch-classification/blob/master/models/cifar/preresnet.py