Fixed issue vlfeat#230 with initBias being ignored.

KingSpencer · Apr 15, 2016 · 43c9f1f · 43c9f1f
1 parent 2487b96
commit 43c9f1f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/imagenet/cnn_imagenet_init.m b/examples/imagenet/cnn_imagenet_init.m
@@ -99,15 +99,17 @@
 end
 convOpts = {'CudnnWorkspaceLimit', opts.cudnnWorkspaceLimit} ;
 net.layers{end+1} = struct('type', 'conv', 'name', sprintf('%s%s', name, id), ...
-                           'weights', {{init_weight(opts, h, w, in, out, 'single'), zeros(out, 1, 'single')}}, ...
+                           'weights', {{init_weight(opts, h, w, in, out, 'single'), ...
+                             ones(out, 1, 'single')*init_bias}}, ...
                            'stride', stride, ...
                            'pad', pad, ...
                            'learningRate', [1 2], ...
                            'weightDecay', [opts.weightDecay 0], ...
                            'opts', {convOpts}) ;
 if opts.batchNormalization
   net.layers{end+1} = struct('type', 'bnorm', 'name', sprintf('bn%s',id), ...
-                             'weights', {{ones(out, 1, 'single'), zeros(out, 1, 'single'), zeros(out, 2, 'single')}}, ...
+                             'weights', {{ones(out, 1, 'single'), zeros(out, 1, 'single'), ...
+                               zeros(out, 2, 'single')}}, ...
                              'learningRate', [2 1 0.05], ...
                              'weightDecay', [0 0]) ;
 end