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

问题描述

我在使用时出错

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

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

请帮忙

解决方法

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

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