machine-learning - 使用我的数据集进行预训练模型的训练 DIGITS 无法正常工作。

Question

我尝试使用预训练模型（VGG 16）对我的数据集进行 DIGITS，该数据集有两个类。但我得到了错误的训练。（我将最后一层的名称更改为 fc8_new） 我上传并成功进入 DIGITS并使用我的数据集进行了测试，该数据集有两个类。
deploy.prototxtVGG_ILSVRC_16_layers.caffemodelsynset_words.txt

Answer 1

Adeploy.prototxt不包括任何损失或准确性层。当您尝试在没有任何损失层的情况下训练网络时，我不确定 Caffe 会做什么。

正如我在这里回答的那样，您将需要创建一个“一体式”网络，其中包括在单个网络中进行训练、验证和部署所需的所有层。@Shai 也链接到这个有用的答案。

正如我们的 GitHub README中所解释的，我们更愿意在我们的用户组上回答这些问题，而不是在 SO 上。