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

Question

我在使用时遇到错误

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

在我将预训练权重 model.ckpt 转换为efficientnetb1_notop.h5

请帮忙

Answer 1

您是在使用带标签的数据还是使用标签预训练的数据？

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