add random seed to base config

msinamsina · Dec 31, 2022 · e6ba56f · e6ba56f
1 parent 81962b4
commit e6ba56f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/configs/POLO/Base-POLO.yaml b/configs/POLO/Base-POLO.yaml
@@ -8,8 +8,8 @@ MODEL:
   FPN:
     IN_FEATURES: ["res2", "res3", "res4", "res5"]
   POLO:
-    SCORE_THR: 0.5
-    UPDATE_THR: 0.5
+    SCORE_THR: 0.1
+    UPDATE_THR: 0.7
 DATASETS:
   TRAIN: ("coco_2017_train",)
   TEST: ("coco_2017_val",)
@@ -21,4 +21,5 @@ SOLVER:
 INPUT:
   MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
   MASK_FORMAT: "bitmask"
+SEED: 1
 VERSION: 2