Hello everyone,
Can someone help with this error? I am running an inception_v3 model via transfer learning with Pytorch, but I receive a CUDNN_STATUS_BAD_PARAM. My input dataset consists of 224x224 patch image (size: [32,3,224,224]). Following is the model structure:
class PretrainedModel(nn.Module):
    def __init__(self, dropout):
        super().__init__()
        model = models.inception_v3(pretrained=True)
        num_ftrs = model.fc.in_features
        model.dropout = dropout
        model.fc = nn.Linear(num_ftrs, 2)
        
        self.model = model

    def forward(self, x):
        return self.model(x)

Bellow is a screenshot of the error I receive:

image.png922×694 38.3 KB

Feedback much appreciated. Thanks