Remove a layer from Inception Model

vision
2

Hi,

  • I hope this may be helpful
     removed = list(Inceptionv3.children())[:-1]
     Inceptionv3 = nn.Sequential(*self.removed)

and here → Use ResNet18 as feature extractor is a question like yours

3 Likes