如何在 DIGITS(不同标签)中使用具有不同数据集的新预训练模型?

How to use new pretraining model with different dataset in DIGITS (different labels)?

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

对于我得到的 VGG 16 层

ERROR: Cannot copy param 0 weights from layer 'fc6'; shape mismatch. Source param shape is 1 1 4096 25088 (102760448); target param shape is 4096 32768 (134217728). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

如何修改图层?

您的标签与 DIGITS 中的数据集相关联 - 而不是您的模型。当您上传预训练的 VGG 模型时,您可能需要重命名最后一个内积层(参见 ),以便您的模型可以在 N 类 而不是 1000.

上运行