解决:网络里的最后一个全连接层,没有设置num_output
坑爹的digits这一句我因为直接用的AlexNet,它注释写的,若把num_output注释掉的话系统会自己设置为你的图片类别数。。结果直接报错,还是得手动设置。。
之后又遇到这个错误,发现主要的原因还是batch_size设大了导致显存不够。。把batch_size调小就行了。
但是为啥用命令行训练的时候256x256的图片,我batch_size设为40都没问题,digits里面设20都出错,用的都是15才行?
解决:网络里的最后一个全连接层,没有设置num_output
坑爹的digits这一句我因为直接用的AlexNet,它注释写的,若把num_output注释掉的话系统会自己设置为你的图片类别数。。结果直接报错,还是得手动设置。。
之后又遇到这个错误,发现主要的原因还是batch_size设大了导致显存不够。。把batch_size调小就行了。
但是为啥用命令行训练的时候256x256的图片,我batch_size设为40都没问题,digits里面设20都出错,用的都是15才行?