冻结TensorFlow对象检测API中的变量



是否有人在食品中使用freeze_variables字段?我试图在配置文件中使用一个正则表达式这样:

freeze_variables: "*FeatureExtractor*"

这样特征提取器的所有变量将被冻结,但我不能,为了我的生命,弄清楚它是否在工作。

我做错了什么吗?如何判断变量是否冻结?tensorboard会给我一些类似的指示吗?

任何指导都将非常感谢

From comments

特别是对于tfood api,我没有正确使用它。的.config文件中的Freeze_variables接受一个模式列表在产生的可训练变量列表中搜索tf.trainable_variables。您可以输入特定层的列表,在在这种情况下,每个条目只有一个匹配,或者您可以输入一个将选择所有匹配的操作名称的模式列表。因此,如果如果你想冻结特征提取器的所有操作,你应该这样做输入["FeatureExtractor"](转述自tom)

有关更多信息,请参阅此处

相关内容

  • 没有找到相关文章

最新更新