如何在DIGITS（不同标签）中使用具有不同数据集的新预训练模型？

Question

我想将VGG_ILSVRC_19_layers用作数字的预训练模型，但具有不同的数据集。 我需要不同的标签文件吗？如何上传此模型并将其用于我的数据集？

我得到的VGG 16层

  

错误：无法复制图层中的参数0权重fc6＆＃39;形状不匹配。   源param形状是1 1 4096 25088（102760448）;目标参数形状   是4096 32768（134217728）。要从中学习此图层的参数   从而不是从保存的网络中复制，重命名该图层。

如何修改图层？

Answer 1

您的标签与DIGITS中的数据集相关联 - 而不是您的模型。当您上传预先训练的VGG模型时，您可能需要重命名最后一个内部产品图层（请参阅this answer），以便您的模型可以使用N个类而不是1000个。