我尝试对 DIGITS 使用预训练模型 (VGG 19),但出现此错误。
ERROR: Your deploy network is missing a Softmax layer! Read the documentation for custom networks and/or look at the standard networks for examples
我尝试使用只有两个类的数据集进行测试。
我读到this和 this尝试修改最后一层,但我也收到错误。如何根据新数据集修改图层?
我尝试修改最后一层,但出现错误
ERROR: Layer 'softmax' references bottom 'fc8' at the TRAIN stage however this blob is not included at that stage. Please consider using an include directive to limit the scope of this layer.
最佳答案
您遇到问题是因为当您确实需要上传“all-in-one”网络时却尝试上传“train/val”网络。不幸的是,我们没有很好地记录这一点。我创建了an RFE提醒我们改进文档。
尝试调整网络中的最后一层,使其看起来像这样:https://github.com/NVIDIA/DIGITS/blob/v4.0.0/digits/standard-networks/caffe/lenet.prototxt#L162-L184
欲了解更多信息,here这就是我建议将 Caffe 的示例网络更新为一体化网络的方式,以及 here这就是我将默认 DIGITS 网络更新为一体化网络的方法。
关于deep-learning - 部署网络缺少 Softmax 层,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40143901/