在预先训练的TensorFlow图表中重新训练最后/顶层的一般方法

时间:2018-04-14 09:07:37

标签: tensorflow object-detection tensorflow-estimator tensorflow-slim

我正在试验我的数据中“Tensorflow detection model zoo”的不同预训练TensorFlow模型 - 我想重新训练(微调)每个模型中最深的N层找到我的数据的最佳配置。 有没有办法以通用形式执行此操作,适用于任何架构,并且不依赖于图层名称(在模型之间更改)? 或者,是否有一种简单的方法可以查询图表中最后N层的名称(我的意思是层 - 而不是操作)?

1 个答案:

答案 0 :(得分:0)

您可以使用training config中的freeze_variables选项来阻止优化较低层。

您想要哪种型号的微调?根据我的经验,冷冻较低的层不能转化为更快的训练或更好的结果(让较低的层漂浮似乎最好)。 YMMV虽然。