首先,我进行微调的时候,代码并非slim提供的train_image_classifier.py,而是自己的代码,我查了slim中的代码,应该是不会出现我文中的问题。
我是在微调完inception模型后,修改代码,想要微调vgg,但是出现如下错误
好象是Prediction出现问题了,追代码,追到vgg的模型定义
其中并没有定义prediction
加上prediction并不会影响网络结构,所以放心加上去
之后就一切正常了