入门深度学习时遇到的问题,小记一下,如大牛有更好的建议,请多多指教。
在调试Inception-v3时出现的error,建议尝试的解决方法如下:
model = torchvision.models.inception_v3(pretrained = True)
model.aux_logits = False
笔者框架使用的是pytorch,其他框架未尝试。
或者尝试:
output, aux = model(input_var)
方法出自:
入门深度学习时遇到的问题,小记一下,如大牛有更好的建议,请多多指教。
在调试Inception-v3时出现的error,建议尝试的解决方法如下:
model = torchvision.models.inception_v3(pretrained = True)
model.aux_logits = False
笔者框架使用的是pytorch,其他框架未尝试。
或者尝试:
output, aux = model(input_var)
方法出自: