首先，我进行微调的时候，代码并非slim提供的train_image_classifier.py，而是自己的代码，我查了slim中的代码，应该是不会出现我文中的问题。
我是在微调完inception模型后，修改代码，想要微调vgg，但是出现如下错误
好象是Prediction出现问题了，追代码，追到vgg的模型定义
其中并没有定义prediction
加上prediction并不会影响网络结构，所以放心加上去
之后就一切正常了
用tf.slim微调vgg模型时遇到的小坑
猜你喜欢
转载自blog.csdn.net/MacwinWin/article/details/78707312
今日推荐
周排行