yolov2: update yolov2 train and solver prototxt

fanghuaqi · fanghuaqi · commit 953a635b8653 · 2018-12-03T07:52:39.000+01:00
Signed-off-by: Huaqi Fang &lt;Huaqi.Fang@synopsys.com&gt;
diff --git a/caffe_models/yolo_v2_voc/caffe_model/yolo_voc_solver.prototxt b/caffe_models/yolo_v2_voc/caffe_model/yolo_voc_solver.prototxt
@@ -1,16 +1,17 @@
 net: "yolo_voc_trainval.prototxt"
 test_iter: 1238
-test_interval: 5000000
+test_interval: 50000
 test_initialization: false
-display: 100
-average_loss: 100
+display: 10
+average_loss: 10
+iter_size: 10
 lr_policy: "step"
 base_lr: 0.00001
-gamma: 0.8
-stepsize: 10000
-max_iter: 100000
+gamma: 0.95
+stepsize: 200
+max_iter: 20000
 momentum: 0.9
 weight_decay: 0.0005
-snapshot: 5000
+snapshot: 500
 snapshot_prefix: "snapshot/"
-#type: "Adam"
+type: "Adam"
diff --git a/caffe_models/yolo_v2_voc/caffe_model/yolo_voc_trainval.prototxt b/caffe_models/yolo_v2_voc/caffe_model/yolo_voc_trainval.prototxt
@@ -28,7 +28,7 @@ layer {
     batch_size: 16
     backend: LMDB
     side: 13
-    prefetch: 3
+    #prefetch: 2
   }
 }
 layer {
@@ -83,7 +83,7 @@ layer {
   bottom: "conv1" 
   top: "bn1"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -139,7 +139,7 @@ layer {
   bottom: "conv2" 
   top: "bn2"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -195,7 +195,7 @@ layer {
   bottom: "conv3" 
   top: "bn3"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -238,7 +238,7 @@ layer {
   bottom: "conv4" 
   top: "bn4"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -282,7 +282,7 @@ layer {
   bottom: "conv5" 
   top: "bn5"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -338,7 +338,7 @@ layer {
   bottom: "conv6" 
   top: "bn6"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -381,7 +381,7 @@ layer {
   bottom: "conv7" 
   top: "bn7"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -425,7 +425,7 @@ layer {
   bottom: "conv8" 
   top: "bn8"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -481,7 +481,7 @@ layer {
   bottom: "conv9" 
   top: "bn9"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -524,7 +524,7 @@ layer {
   bottom: "conv10" 
   top: "bn10"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -568,7 +568,7 @@ layer {
   bottom: "conv11" 
   top: "bn11"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -611,7 +611,7 @@ layer {
   bottom: "conv12" 
   top: "bn12"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -655,7 +655,7 @@ layer {
   bottom: "conv13" 
   top: "bn13"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -711,7 +711,7 @@ layer {
   bottom: "conv14" 
   top: "bn14"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -754,7 +754,7 @@ layer {
   bottom: "conv15" 
   top: "bn15"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -798,7 +798,7 @@ layer {
   bottom: "conv16" 
   top: "bn16"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -841,7 +841,7 @@ layer {
   bottom: "conv17" 
   top: "bn17"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -885,7 +885,7 @@ layer {
   bottom: "conv18" 
   top: "bn18"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -929,7 +929,7 @@ layer {
   bottom: "conv19" 
   top: "bn19"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -973,7 +973,7 @@ layer {
   bottom: "conv20" 
   top: "bn20"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -1023,7 +1023,7 @@ layer {
   bottom: "conv22" 
   top: "bn22"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {
@@ -1084,7 +1084,7 @@ layer {
   bottom: "conv25" 
   top: "bn25"
   batch_norm_param {
-    use_global_stats: false
+    #use_global_stats: false
     }
   }
 layer {

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ layer {`
`28`	`28`	`batch_size: 16`
`29`	`29`	`backend: LMDB`
`30`	`30`	`side: 13`
`31`		`- prefetch: 3`
	`31`	`+ #prefetch: 2`
`32`	`32`	`}`
`33`	`33`	`}`
`34`	`34`	`layer {`
`@@ -83,7 +83,7 @@ layer {`
`83`	`83`	`bottom: "conv1"`
`84`	`84`	`top: "bn1"`
`85`	`85`	`batch_norm_param {`
`86`		`- use_global_stats: false`
	`86`	`+ #use_global_stats: false`
`87`	`87`	`}`
`88`	`88`	`}`
`89`	`89`	`layer {`
`@@ -139,7 +139,7 @@ layer {`
`139`	`139`	`bottom: "conv2"`
`140`	`140`	`top: "bn2"`
`141`	`141`	`batch_norm_param {`
`142`		`- use_global_stats: false`
	`142`	`+ #use_global_stats: false`
`143`	`143`	`}`
`144`	`144`	`}`
`145`	`145`	`layer {`
`@@ -195,7 +195,7 @@ layer {`
`195`	`195`	`bottom: "conv3"`
`196`	`196`	`top: "bn3"`
`197`	`197`	`batch_norm_param {`
`198`		`- use_global_stats: false`
	`198`	`+ #use_global_stats: false`
`199`	`199`	`}`
`200`	`200`	`}`
`201`	`201`	`layer {`
`@@ -238,7 +238,7 @@ layer {`
`238`	`238`	`bottom: "conv4"`
`239`	`239`	`top: "bn4"`
`240`	`240`	`batch_norm_param {`
`241`		`- use_global_stats: false`
	`241`	`+ #use_global_stats: false`
`242`	`242`	`}`
`243`	`243`	`}`
`244`	`244`	`layer {`
`@@ -282,7 +282,7 @@ layer {`
`282`	`282`	`bottom: "conv5"`
`283`	`283`	`top: "bn5"`
`284`	`284`	`batch_norm_param {`
`285`		`- use_global_stats: false`
	`285`	`+ #use_global_stats: false`
`286`	`286`	`}`
`287`	`287`	`}`
`288`	`288`	`layer {`
`@@ -338,7 +338,7 @@ layer {`
`338`	`338`	`bottom: "conv6"`
`339`	`339`	`top: "bn6"`
`340`	`340`	`batch_norm_param {`
`341`		`- use_global_stats: false`
	`341`	`+ #use_global_stats: false`
`342`	`342`	`}`
`343`	`343`	`}`
`344`	`344`	`layer {`
`@@ -381,7 +381,7 @@ layer {`
`381`	`381`	`bottom: "conv7"`
`382`	`382`	`top: "bn7"`
`383`	`383`	`batch_norm_param {`
`384`		`- use_global_stats: false`
	`384`	`+ #use_global_stats: false`
`385`	`385`	`}`
`386`	`386`	`}`
`387`	`387`	`layer {`
`@@ -425,7 +425,7 @@ layer {`
`425`	`425`	`bottom: "conv8"`
`426`	`426`	`top: "bn8"`
`427`	`427`	`batch_norm_param {`
`428`		`- use_global_stats: false`
	`428`	`+ #use_global_stats: false`
`429`	`429`	`}`
`430`	`430`	`}`
`431`	`431`	`layer {`
`@@ -481,7 +481,7 @@ layer {`
`481`	`481`	`bottom: "conv9"`
`482`	`482`	`top: "bn9"`
`483`	`483`	`batch_norm_param {`
`484`		`- use_global_stats: false`
	`484`	`+ #use_global_stats: false`
`485`	`485`	`}`
`486`	`486`	`}`
`487`	`487`	`layer {`
`@@ -524,7 +524,7 @@ layer {`
`524`	`524`	`bottom: "conv10"`
`525`	`525`	`top: "bn10"`
`526`	`526`	`batch_norm_param {`
`527`		`- use_global_stats: false`
	`527`	`+ #use_global_stats: false`
`528`	`528`	`}`
`529`	`529`	`}`
`530`	`530`	`layer {`
`@@ -568,7 +568,7 @@ layer {`
`568`	`568`	`bottom: "conv11"`
`569`	`569`	`top: "bn11"`
`570`	`570`	`batch_norm_param {`
`571`		`- use_global_stats: false`
	`571`	`+ #use_global_stats: false`
`572`	`572`	`}`
`573`	`573`	`}`
`574`	`574`	`layer {`
`@@ -611,7 +611,7 @@ layer {`
`611`	`611`	`bottom: "conv12"`
`612`	`612`	`top: "bn12"`
`613`	`613`	`batch_norm_param {`
`614`		`- use_global_stats: false`
	`614`	`+ #use_global_stats: false`
`615`	`615`	`}`
`616`	`616`	`}`
`617`	`617`	`layer {`
`@@ -655,7 +655,7 @@ layer {`
`655`	`655`	`bottom: "conv13"`
`656`	`656`	`top: "bn13"`
`657`	`657`	`batch_norm_param {`
`658`		`- use_global_stats: false`
	`658`	`+ #use_global_stats: false`
`659`	`659`	`}`
`660`	`660`	`}`
`661`	`661`	`layer {`
`@@ -711,7 +711,7 @@ layer {`
`711`	`711`	`bottom: "conv14"`
`712`	`712`	`top: "bn14"`
`713`	`713`	`batch_norm_param {`
`714`		`- use_global_stats: false`
	`714`	`+ #use_global_stats: false`
`715`	`715`	`}`
`716`	`716`	`}`
`717`	`717`	`layer {`
`@@ -754,7 +754,7 @@ layer {`
`754`	`754`	`bottom: "conv15"`
`755`	`755`	`top: "bn15"`
`756`	`756`	`batch_norm_param {`
`757`		`- use_global_stats: false`
	`757`	`+ #use_global_stats: false`
`758`	`758`	`}`
`759`	`759`	`}`
`760`	`760`	`layer {`
`@@ -798,7 +798,7 @@ layer {`
`798`	`798`	`bottom: "conv16"`
`799`	`799`	`top: "bn16"`
`800`	`800`	`batch_norm_param {`
`801`		`- use_global_stats: false`
	`801`	`+ #use_global_stats: false`
`802`	`802`	`}`
`803`	`803`	`}`
`804`	`804`	`layer {`
`@@ -841,7 +841,7 @@ layer {`
`841`	`841`	`bottom: "conv17"`
`842`	`842`	`top: "bn17"`
`843`	`843`	`batch_norm_param {`
`844`		`- use_global_stats: false`
	`844`	`+ #use_global_stats: false`
`845`	`845`	`}`
`846`	`846`	`}`
`847`	`847`	`layer {`
`@@ -885,7 +885,7 @@ layer {`
`885`	`885`	`bottom: "conv18"`
`886`	`886`	`top: "bn18"`
`887`	`887`	`batch_norm_param {`
`888`		`- use_global_stats: false`
	`888`	`+ #use_global_stats: false`
`889`	`889`	`}`
`890`	`890`	`}`
`891`	`891`	`layer {`
`@@ -929,7 +929,7 @@ layer {`
`929`	`929`	`bottom: "conv19"`
`930`	`930`	`top: "bn19"`
`931`	`931`	`batch_norm_param {`
`932`		`- use_global_stats: false`
	`932`	`+ #use_global_stats: false`
`933`	`933`	`}`
`934`	`934`	`}`
`935`	`935`	`layer {`
`@@ -973,7 +973,7 @@ layer {`
`973`	`973`	`bottom: "conv20"`
`974`	`974`	`top: "bn20"`
`975`	`975`	`batch_norm_param {`
`976`		`- use_global_stats: false`
	`976`	`+ #use_global_stats: false`
`977`	`977`	`}`
`978`	`978`	`}`
`979`	`979`	`layer {`
`@@ -1023,7 +1023,7 @@ layer {`
`1023`	`1023`	`bottom: "conv22"`
`1024`	`1024`	`top: "bn22"`
`1025`	`1025`	`batch_norm_param {`
`1026`		`- use_global_stats: false`
	`1026`	`+ #use_global_stats: false`
`1027`	`1027`	`}`
`1028`	`1028`	`}`
`1029`	`1029`	`layer {`
`@@ -1084,7 +1084,7 @@ layer {`
`1084`	`1084`	`bottom: "conv25"`
`1085`	`1085`	`top: "bn25"`
`1086`	`1086`	`batch_norm_param {`
`1087`		`- use_global_stats: false`
	`1087`	`+ #use_global_stats: false`
`1088`	`1088`	`}`
`1089`	`1089`	`}`
`1090`	`1090`	`layer {`