doc: sync with latest changes to the DAG converter

minsucho · Oct 12, 2015 · 4d40a19 · 4d40a19
1 parent 9308c75
commit 4d40a19
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/doc/site/docs/quick.md b/doc/site/docs/quick.md
@@ -73,7 +73,7 @@ im_ = im_ - net.meta.normalization.averageImage ;
 net.eval({'data', im_}) ;
 
 % obtain the CNN otuput
-scores = net.vars(net.getVarIndex('prediction')).value ;
+scores = net.vars(net.getVarIndex('prob')).value ;
 scores = squeeze(gather(scores)) ;
 
 % show the classification results

diff --git a/examples/cnn_imagenet_camdemo.m b/examples/cnn_imagenet_camdemo.m
@@ -18,7 +18,7 @@ function cnn_imagenet_camdemo()
 
 if strcmp(model, 'imagenet-googlenet-dag')
   net = dagnn.DagNN.loadobj(net) ;
-  out = net.getVarIndex('prediction') ;
+  out = net.getVarIndex('prob') ;
   normalization = net.meta.normalization ;
   description = net.meta.classes.description ;
   dag = true ;