fix config yaml bug

OrangePeeling · Aug 20, 2020 · b2b070b · b2b070b
1 parent be110c1
commit b2b070b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/config/tusimple_lanenet.yaml b/config/tusimple_lanenet.yaml
@@ -80,6 +80,10 @@ SOLVER:
     WEIGHT_DECAY: 0.0005
     MOVING_AVE_DECAY: 0.9995
     LOSS_TYPE: 'cross_entropy'
+    OHEM:
+        ENABLE: False
+        SCORE_THRESH: 0.65
+        MIN_SAMPLE_NUMS: 65536
 GPU:
     GPU_MEMORY_FRACTION: 0.9
     TF_ALLOW_GROWTH: True