2

I try to use pretrained model (VGG 19) to DIGITS but I got this error.

ERROR: Your deploy network is missing a Softmax layer! Read the documentation for custom networks and/or look at the standard networks for examples

I try to test with my dataset which has only two classes.

I read this and this try to modify last layer but also I got error. How can I modify layers based on new dataset?

I try to modify the last layer and I got error

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.

Community
  • 1
  • 1
Video Analysis Deep Learning
  • 338
  • 2
  • 12

1 Answers1

2

You're having a problem because you're trying to upload a "train/val" network when you really need to be uploading an "all-in-one" network. Unfortunately, we don't document this very well. I've created an RFE to remind us to improve the documentation.

Try to adjust the last layers in your network to look something like this: https://github.com/NVIDIA/DIGITS/blob/v4.0.0/digits/standard-networks/caffe/lenet.prototxt#L162-L184

For more information, here is how I've proposed updating Caffe's example networks to all-in-one nets, and here is how I updated the default DIGITS networks to be all-in-one nets.

Luke Yeager
  • 1,400
  • 1
  • 17
  • 30
  • Thanks but I got error when I change the last layer. "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." How can I change the last layer? – Video Analysis Deep Learning Oct 20 '16 at 01:36
  • How can I change the last layer? How can I use "all-in-one"? The "all-in-one" related to Fine-tuning Models? for using VGG 19 layers how can I modify last layer in order to use in my dataset which has only two classes? – Video Analysis Deep Learning Oct 20 '16 at 01:47
  • 1
    you might find [this answer](http://stackoverflow.com/a/33773152/1714410) useful: how to convert "train" prototxt to "deploy" prototxt. – Shai Oct 20 '16 at 04:26
  • How can I create All in one network? DIGITS support all-in-one network? – Video Analysis Deep Learning Dec 23 '16 at 07:18