finetune时,alexnet出现输入与目标不一致RuntimeError: invalid argument 2: size '[64 x 9216]' is invalid for input of with,是由于
/jupyter/jupyter_env/local/lib/python2.7/site-packages/torchvision/models/alexnet.pyc的设置成针对imagenet的
x = x.view(x.size(0),256*6*6),所以在训练别的大小的图像时,需要改为x = x.view(x.size(0),-1),然后删除/alexnet.pyc文件后,使用
import
py_compile
py_compile.
compile
(r
'路径\ASA.py'
)
将python文件转化为pyc文件