我想在DIGITS中使用VGG_ILSVRC_19_layers作为预训练模型，但使用不同的数据集。我是否需要不同的标签文件？我怎样才能上传这个模型并将其用于我的数据集？

对于VGG 16层，我得到了以下错误：

错误：无法从层’fc6’复制参数0的权重；形状不匹配。源参数形状为1 1 4096 25088（102760448）；目标参数形状为4096 32768（134217728）。要从头开始学习这一层的参数，而不是从保存的网络中复制，请重命名该层。

如何修改这些层？

回答：

在DIGITS中，您的标签与您的数据集相关联，而不是与您的模型。当您上传预训练的VGG模型时，您可能需要重命名最后一个内积层（参见这个回答），这样您的模型就可以处理N个类别而不是1000个类别。