我想使用INCEPTION_V4模型作为特征提取器进行转移学习，并从Tensorflow的GitHub存储库下载了代码和检查点文件
然后，我在5个类中添加了自己的分类层。但是在使用tf.train.Saver进行模型还原期间，它显示错误，即找不到添加的图层的变量值。
为了解决这个问题，我创建了两个单独的图，一个用于加载预训练的模型，另一个用于我的分类层。但是我不能将一个图的输出作为第二个图的输入。您能建议其他方法来进行转学或解决问题吗
一种简单的解决方案是仅构建初始模型（不包含您的图层），然后创建保护程序并将其用于还原，然后再创建您的图层。
还有更复杂的解决方案（您可以将的var_list参数tf.Saver与要还原的所有变量的列表一起传递，并将其初始化为所有初始变量的列表），但是这应该很简单，我用我的方法来完成转移学习模型。
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句