Tensorflow，問題加載檢查點以通過微調進行對象檢測[在檢查點中不可用]

Question

我一直在嘗試加載在openimages上預先訓練的“ faster_rcnn_inception_resnet_v2_atrous”檢查點。 我收到一條錯誤消息，指出檢查點中缺少某些值。 鑒於此，我已經使用其他模型，但沒有收到此錯誤。

我在模型上運行了inspect_checkpoint.py，發現實際上確實缺少零件。 這是錯誤日志的一部分。

W0204 17：09：30.737426 3724 variables_helper.py:144]變量[FirstStageBoxPredictor / BoxEncodingPredictor / biases / Momentum]在檢查點不可用W0204 17：09：30.738423 3724 variables_helper.py:144]變量[FirstStageBoxPredictor / BoxEncodingPredictor / weights / Momentum ]在檢查點W0204中不可用17：09：30.738423 3724 variables_helper.py:144]變量[FirstStageBoxPredictor / ClassPredictor / biases / Momentum]在檢查點W0204中不可用17：09：30.739422 3724 variables_helper.py:144]變量[FirstStageBoxPredictor / ClassPredictor / weights / Momentum]在檢查點W0204中不可用17：09：30.740419 3724 variables_helper.py:144]變量[FirstStageFeatureExtractor / InceptionResnetV2 / Conv2d_1a_3x3 / BatchNorm / beta / Momentum]在檢查點W0204 17：09：30741中不可用。 3724 variables_helper.py:144]變量[FirstStageFeatureExtractor / InceptionResnetV2 / Conv2d_1a_3x3 / weights / Momentum]在檢查點不可用

基本上所有末尾帶有“動量”的值都將丟失。 有想法該怎么解決這個嗎？ 謝謝！

Answer 1

這不是問題，動量值未保存到模型中，但是由於您正在微調（而不是從同一位置繼續訓練），因此您不想加載和使用這些值，可以將它們設為初始化。