Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commit96dc9f1

Browse files
committed
Force LRN engine to CAFFE (fixes major perf bug when engine defaults to CUDNN)
1 parent8ff49d3 commit96dc9f1

File tree

10 files changed

+20
-0
lines changed

10 files changed

+20
-0
lines changed

‎models/pascal_voc/ZF/fast_rcnn/test.prototxt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ layer {
4444
alpha: 0.00005
4545
beta: 0.75
4646
norm_region: WITHIN_CHANNEL
47+
engine: CAFFE
4748
}
4849
}
4950
layer {
@@ -86,6 +87,7 @@ layer {
8687
alpha: 0.00005
8788
beta: 0.75
8889
norm_region: WITHIN_CHANNEL
90+
engine: CAFFE
8991
}
9092
}
9193
layer {

‎models/pascal_voc/ZF/fast_rcnn/train.prototxt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ layer {
4747
alpha: 0.00005
4848
beta: 0.75
4949
norm_region: WITHIN_CHANNEL
50+
engine: CAFFE
5051
}
5152
}
5253
layer {
@@ -91,6 +92,7 @@ layer {
9192
alpha: 0.00005
9293
beta: 0.75
9394
norm_region: WITHIN_CHANNEL
95+
engine: CAFFE
9496
}
9597
}
9698
layer {

‎models/pascal_voc/ZF/faster_rcnn_alt_opt/faster_rcnn_test.pt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ layer {
4444
alpha: 0.00005
4545
beta: 0.75
4646
norm_region: WITHIN_CHANNEL
47+
engine: CAFFE
4748
}
4849
}
4950
layer {
@@ -86,6 +87,7 @@ layer {
8687
alpha: 0.00005
8788
beta: 0.75
8889
norm_region: WITHIN_CHANNEL
90+
engine: CAFFE
8991
}
9092
}
9193
layer {

‎models/pascal_voc/ZF/faster_rcnn_alt_opt/rpn_test.pt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ layer {
4343
alpha: 0.00005
4444
beta: 0.75
4545
norm_region: WITHIN_CHANNEL
46+
engine: CAFFE
4647
}
4748
}
4849
layer {
@@ -86,6 +87,7 @@ layer {
8687
alpha: 0.00005
8788
beta: 0.75
8889
norm_region: WITHIN_CHANNEL
90+
engine: CAFFE
8991
}
9092
}
9193
layer {

‎models/pascal_voc/ZF/faster_rcnn_alt_opt/stage1_fast_rcnn_train.pt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ layer {
4747
alpha: 0.00005
4848
beta: 0.75
4949
norm_region: WITHIN_CHANNEL
50+
engine: CAFFE
5051
}
5152
}
5253
layer {
@@ -91,6 +92,7 @@ layer {
9192
alpha: 0.00005
9293
beta: 0.75
9394
norm_region: WITHIN_CHANNEL
95+
engine: CAFFE
9496
}
9597
}
9698
layer {

‎models/pascal_voc/ZF/faster_rcnn_alt_opt/stage1_rpn_train.pt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ layer {
4444
alpha: 0.00005
4545
beta: 0.75
4646
norm_region: WITHIN_CHANNEL
47+
engine: CAFFE
4748
}
4849
}
4950
layer {
@@ -88,6 +89,7 @@ layer {
8889
alpha: 0.00005
8990
beta: 0.75
9091
norm_region: WITHIN_CHANNEL
92+
engine: CAFFE
9193
}
9294
}
9395
layer {

‎models/pascal_voc/ZF/faster_rcnn_alt_opt/stage2_fast_rcnn_train.pt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ layer {
4747
alpha: 0.00005
4848
beta: 0.75
4949
norm_region: WITHIN_CHANNEL
50+
engine: CAFFE
5051
}
5152
}
5253
layer {
@@ -91,6 +92,7 @@ layer {
9192
alpha: 0.00005
9293
beta: 0.75
9394
norm_region: WITHIN_CHANNEL
95+
engine: CAFFE
9496
}
9597
}
9698
layer {

‎models/pascal_voc/ZF/faster_rcnn_alt_opt/stage2_rpn_train.pt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ layer {
4444
alpha: 0.00005
4545
beta: 0.75
4646
norm_region: WITHIN_CHANNEL
47+
engine: CAFFE
4748
}
4849
}
4950
layer {
@@ -88,6 +89,7 @@ layer {
8889
alpha: 0.00005
8990
beta: 0.75
9091
norm_region: WITHIN_CHANNEL
92+
engine: CAFFE
9193
}
9294
}
9395
layer {

‎models/pascal_voc/ZF/faster_rcnn_end2end/test.prototxt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ layer {
4444
alpha: 0.00005
4545
beta: 0.75
4646
norm_region: WITHIN_CHANNEL
47+
engine: CAFFE
4748
}
4849
}
4950
layer {
@@ -86,6 +87,7 @@ layer {
8687
alpha: 0.00005
8788
beta: 0.75
8889
norm_region: WITHIN_CHANNEL
90+
engine: CAFFE
8991
}
9092
}
9193
layer {

‎models/pascal_voc/ZF/faster_rcnn_end2end/train.prototxt‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ layer {
4444
alpha: 0.00005
4545
beta: 0.75
4646
norm_region: WITHIN_CHANNEL
47+
engine: CAFFE
4748
}
4849
}
4950
layer {
@@ -88,6 +89,7 @@ layer {
8889
alpha: 0.00005
8990
beta: 0.75
9091
norm_region: WITHIN_CHANNEL
92+
engine: CAFFE
9193
}
9294
}
9395
layer {

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp