我正在试验我的数据中“Tensorflow detection model zoo”的不同预训练TensorFlow模型 - 我想重新训练(微调)每个模型中最深的N层找到我的数据的最佳配置。 有没有办法以通用形式执行此操作,适用于任何架构,并且不依赖于图层名称(在模型之间更改)? 或者,是否有一种简单的方法可以查询图表中最后N层的名称(我的意思是层 - 而不是操作)?
答案 0 :(得分:0)
您可以使用training config中的freeze_variables
选项来阻止优化较低层。
您想要哪种型号的微调?根据我的经验,冷冻较低的层不能转化为更快的训练或更好的结果(让较低的层漂浮似乎最好)。 YMMV虽然。