fix bug for plain net

zhanghang1989 · Sep 15, 2016 · 01d2790 · 01d2790
1 parent 8e3a4ab
commit 01d2790
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
 data/
+*.swp
diff --git a/init/res_cifar_init.m b/init/res_cifar_init.m
@@ -12,7 +12,7 @@
 
 
 opts.networkType = 'resnet'; % 'plain' | 'resnet'
-opts.reLUafterSum = false;
+opts.reLUafterSum = true;
 opts.shortcutBN = false;
 opts = vl_argparse(opts, varargin); 
 
@@ -85,12 +85,12 @@
     lName = sprintf('relu%d', info.lastIdx);
   end
   add_block_conv(net, sprintf('%d', info.lastIdx+1), lName, ...
-    [w w info.lastNumChannel ch], stride, bn, true); 
+    [w w info.lastNumChannel ch], stride, opts); 
   info.lastIdx = info.lastIdx + 1;
   info.lastNumChannel = ch;
   for i=2:2*n,
     add_block_conv(net, sprintf('%d', info.lastIdx+1), sprintf('relu%d', info.lastIdx), ...
-      [w w ch ch], 1, bn, true);
+      [w w ch ch], 1, opts);
     info.lastIdx = info.lastIdx + 1;
   end
 elseif strcmpi(netType, 'resnet'), 

diff --git a/res_cifar.m b/res_cifar.m
@@ -4,7 +4,7 @@
 setup;
 opts.modelType = 'resnet' ;
 opts.preActivation = false;
-opts.reLUafterSum = false;
+opts.reLUafterSum = true;
 opts.shortcutBN = false;
 [opts, varargin] = vl_argparse(opts, varargin) ;