network-slimming icon indicating copy to clipboard operation
network-slimming copied to clipboard

Published 20 hours ago • verified publisher icon Eric-mingjie

Question about resnet50

Open ZhiweiYan-96 opened this issue 5 years ago • 2 comments

Hi Eric, Since I need to train a ResNet50 with less parameter, i read the code of definition of ResNet in your code. I found that it is pretty different from official code of Pytorch torchvision model. Could you please tell how i can change the code so i can train a pruned ResNet50?

ZhiweiYan-96 avatar Apr 27 '19 12:04 ZhiweiYan-96

The official torchvision contains the ResNet model for ImageNet dataset. While our code is for CIFAR-10 and CIFAR-100, this is why the model definition is different.

Eric-mingjie avatar Apr 27 '19 12:04 Eric-mingjie

Hi Eric, Since I need to train a ResNet50 with less parameter, i read the code of definition of ResNet in your code. I found that it is pretty different from official code of Pytorch torchvision model. Could you please tell how i can change the code so i can train a pruned ResNet50?

Hi,have you trained it in ResNet50?

tanbo1 avatar Jul 08 '19 02:07 tanbo1