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

这不是问题，动量值未保存到模型中，但是由于您正在微调（而不是从同一位置继续训练），因此您不想加载和使用这些值，可以将它们设为初始化。