AttributeError: 'InceptionOutputs' object has no attribute 'log_softmax'

Pytorch 加载InceptionV3预训练模型报错

只需修改训练模型时的代码:

outputs,aux2= self.model(inputs)  

即可

 

相关文章: