Force LRN engine to CAFFE (fixes major perf bug when engine defaults …

…to CUDNN)
shls · Mar 8, 2016 · 96dc9f1 · 96dc9f1
1 parent 8ff49d3
commit 96dc9f1
Show file tree

Hide file tree

Showing 10 changed files with 20 additions and 0 deletions.
diff --git a/models/pascal_voc/ZF/fast_rcnn/test.prototxt b/models/pascal_voc/ZF/fast_rcnn/test.prototxt
@@ -44,6 +44,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -86,6 +87,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/fast_rcnn/train.prototxt b/models/pascal_voc/ZF/fast_rcnn/train.prototxt
@@ -47,6 +47,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -91,6 +92,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/faster_rcnn_alt_opt/faster_rcnn_test.pt b/models/pascal_voc/ZF/faster_rcnn_alt_opt/faster_rcnn_test.pt
@@ -44,6 +44,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -86,6 +87,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/faster_rcnn_alt_opt/rpn_test.pt b/models/pascal_voc/ZF/faster_rcnn_alt_opt/rpn_test.pt
@@ -43,6 +43,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -86,6 +87,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/faster_rcnn_alt_opt/stage1_fast_rcnn_train.pt b/models/pascal_voc/ZF/faster_rcnn_alt_opt/stage1_fast_rcnn_train.pt
@@ -47,6 +47,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -91,6 +92,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/faster_rcnn_alt_opt/stage1_rpn_train.pt b/models/pascal_voc/ZF/faster_rcnn_alt_opt/stage1_rpn_train.pt
@@ -44,6 +44,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -88,6 +89,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/faster_rcnn_alt_opt/stage2_fast_rcnn_train.pt b/models/pascal_voc/ZF/faster_rcnn_alt_opt/stage2_fast_rcnn_train.pt
@@ -47,6 +47,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -91,6 +92,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/faster_rcnn_alt_opt/stage2_rpn_train.pt b/models/pascal_voc/ZF/faster_rcnn_alt_opt/stage2_rpn_train.pt
@@ -44,6 +44,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -88,6 +89,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/faster_rcnn_end2end/test.prototxt b/models/pascal_voc/ZF/faster_rcnn_end2end/test.prototxt
@@ -44,6 +44,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -86,6 +87,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {

diff --git a/models/pascal_voc/ZF/faster_rcnn_end2end/train.prototxt b/models/pascal_voc/ZF/faster_rcnn_end2end/train.prototxt
@@ -44,6 +44,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {
@@ -88,6 +89,7 @@ layer {
 		alpha: 0.00005
 		beta: 0.75
 		norm_region: WITHIN_CHANNEL
+    engine: CAFFE
 	}
 }
 layer {