I’m trying to train a pre-trained Inception v3 model for my task, which gives as input 178x178 images. It has 5 possible classes so I changed the fully-connected layer to have 5 output feature. My code is the following: # Pre-trained models model = models.inception_v3(pretrained=True) ### ResNet …

Inception v3 pre-trained model

Hasith_Karunasekera (Hasith Karunasekera) March 16, 2022, 2:13pm 8

This solution worked;

model = models.inception_v3(pretrained=True)
model.aux_logits = False

Found the solution here;