Adds missing config option in Faster's base config

efernandez · Oct 10, 2017 · 78b24e7 · 78b24e7
1 parent 133f23f
commit 78b24e7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/luminoth/models/fasterrcnn/base_config.yml b/luminoth/models/fasterrcnn/base_config.yml
@@ -181,7 +181,8 @@ rcnn:
   layer_sizes: [4096, 4096]
   # Dropout keep probability (turn off with 1.0 value)
   dropout_keep_prop: 1.0
-  activation_function: relu6
+  activation_function: relu
+  use_mean: False
   initializer:
     type: variance_scaling_initializer
     factor: 1.0