要冻结哪些InceptionResNetV2层以进行微调

时间:2018-10-03 09:01:47

标签: neural-network keras deep-learning

我正在尝试通过以下过程微调我的深度学习模型:

  1. 将经过预训练的模型作为基础模型加载,并删除顶层。
  2. 冻结基本模型的所有层。
  3. 在顶部添加自定义密集层。
  4. 对网络进行一些训练,现在已经建立了密集层。
  5. 解冻顶部卷积层块(通常是顶部1或2)
  6. 这次再次训练密集层,并与未冻结的卷积层一起训练,并使用具有较低学习率的优化程序。

对于诸如InceptionV3和ResNet之类的模型,不难知道要在第5步中解冻哪些块。人们可以向上查找层数,或者通过手动检查得出层数。

但是,当我查看InceptionResNetV2时,我正在努力进行微调,我需要解冻哪些层。对我而言,模型的体系结构和打印摘要时的块命名对我来说并不直观。

0 个答案:

没有答案