VGG16 Model Outputs Incorrect dimension - Transfer Learning

前端 未结 0 1476
北荒
北荒 2021-01-24 04:13

I am trying to use a pre-trained VGG16 model to classify CIFAR10 on pyTorch. The model was originally trained on ImageNet.

Here is how I imported and modified the model:<

相关标签:
回答
  • 消灭零回复
提交回复
热议问题