pytorch-cnn-finetune

pytorch-cnn-finetune copied to clipboard

When I am trying to run examples/cifar10.py with Inception-Resnet-v2, i am getting the following error: RuntimeError: Calculated padded input size per channel: (1 x 1). Kernel size: (3 x 3)....

roy-shudipto

how can i load the trained-model

2

hello, i finetune the pre-trained inception_v4 model with 3 classes,by model=make_model('inception_v4',3,pretrained=True) , and saved the trained-model as 'model.pth' by torch.save(model.state_dict(),'model.pth') but when i restored the model by model = make_model('inception_v4',3,pretrained=False)...

LianShuaiLong

What does your network “output ” include? “output = model(data) ” Is “output.data () ” a logical value？ I want to get the classification probability, how do I get it?...

liweizhong666

Auto Flattening of features

1

Hi, I found this annoying feature/bug as one may like to: ``` def forward(self, x): x = self.features(x) if self.pool is not None: x = self.pool(x) if self.dropout is not...

Geeks-Sid