pytorch问题解决

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文件