我尝试使用预训练模型(VGG 16)在DIGITS中训练我的数据集,该数据集包含两个类别,但训练结果出现错误。(我将最后一层的名称更改为fc8_new)
我成功上传了deploy.prototxt
、VGG_ILSVRC_16_layers.caffemodel
和synset_words.txt
到DIGITS,并使用包含两个类别的数据集进行了测试。
回答:
deploy.prototxt
不会包含任何损失或准确率层。我不太确定当你尝试训练一个没有损失层的网络时,Caffe会做些什么。
正如我在这里回答的那样,你需要创建一个“全合一”的网络,该网络在一个网络中包含训练、验证和部署所需的所有层。@某人链接到了这个有用的回答。
正如在我们的GitHub README中所解释的,我们更希望在我们的用户组而不是在SO上回答这些问题。