如何在 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.
上运行
我想使用 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 模型时,您可能需要重命名最后一个内积层(参见