xavier weight init now default in MLP template for Caffe

revilokeb · Aug 10, 2015 · 02c7a6b · 02c7a6b
1 parent 2e6dbfe
commit 02c7a6b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/templates/caffe/mlp/deploy.prototxt b/templates/caffe/mlp/deploy.prototxt
@@ -19,7 +19,7 @@ layer {
   inner_product_param {
     num_output: 50
     weight_filler {
-      type: "gaussian"
+      type: "xavier"
       std: 0.1
     }
     bias_filler {
@@ -50,7 +50,7 @@ layer {
   inner_product_param {
     num_output: 20
     weight_filler {
-      type: "gaussian"
+      type: "xavier"
       std: 0.1
     }
     bias_filler {

diff --git a/templates/caffe/mlp/mlp.prototxt b/templates/caffe/mlp/mlp.prototxt
@@ -33,7 +33,7 @@ layer {
   inner_product_param {
     num_output: 50
     weight_filler {
-      type: "gaussian"
+      type: "xavier"
       std: 0.1
     }
     bias_filler {
@@ -64,7 +64,7 @@ layer {
   inner_product_param {
     num_output: 20
     weight_filler {
-      type: "gaussian"
+      type: "xavier"
       std: 0.1
     }
     bias_filler {