如何加载.pb protobuf模型,然后根据需要调整网络(特别是外层),以便为完全不同的类训练新模型?有效地进行迁移学习?
我想做一些类似于these(即训练外层的学习率比内层的学习率高)的事情,因此我需要一种方法,不仅加载带有变量的图,而且还更改网络的结构和超参数。
如果有人可以效仿《盗梦空间》的模式,那就太棒了!
我的问题与this one非常相似。
我在互联网上到处搜索(TF docs、Github、StackOverflow、Google……)但是我找不到对新手有用的东西。
谢谢!
最佳答案
这是来自Tensorflow官方网站https://www.tensorflow.org/hub/tutorials/image_retraining的更新教程
他们使用预先训练过的Inception V3模型,一切正常。您可以将数据集文件夹更改为自己的数据集