EfficientnetB1:ValueError:您正在尝试将包含185层的权重文件加载到具有184层的模型中.<

Question

当我使用

时出现错误

model = EfficientNetB1(weights="efficientnetb1_notop.h5", include_top=False)

后，我转换了预训练的权重模型。CKPT到efficientnetb1_notop.h5

请帮

Answer 1

您是使用标记数据还是使用标签预训练的数据?

include_top=True完全按原样部署模型。