问题说明:
今天用torchvision.models.inception_v3使用inception_v3的时候,训练时报错。
解决方法:
把net = torchvision.models.inception_v3(pretrained=False).to(DEVICE)
改成:
net = torchvision.models.inception_v3(aux_logits=False,pretrained=False).to(DEVICE)
今天用torchvision.models.inception_v3使用inception_v3的时候,训练时报错。
把net = torchvision.models.inception_v3(pretrained=False).to(DEVICE)
改成:
net = torchvision.models.inception_v3(aux_logits=False,pretrained=False).to(DEVICE)