diff --git a/examples/cnn_imagenet.m b/examples/cnn_imagenet.m index 562ec5a3..69131709 100644 --- a/examples/cnn_imagenet.m +++ b/examples/cnn_imagenet.m @@ -59,7 +59,12 @@ function cnn_imagenet(varargin) net = cnn_imagenet_init('model', opts.modelType, ... 'batchNormalization', opts.batchNormalization, ... 'weightInitMethod', opts.weightInitMethod) ; -bopts = net.normalization ; +switch lower(opts.networkType) + case 'simplenn' + bopts = net.normalization ; + case 'dagnn' + bopts = net.meta.normalization ; +end bopts.numThreads = opts.numFetchThreads ; % compute image statistics (mean, RGB covariances etc)