Update configs

stefanklut · Apr 19, 2024 · 11b14b5 · 11b14b5
1 parent 094f715
commit 11b14b5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/configs/segmentation/baseline/baseline_general.yaml b/configs/segmentation/baseline/baseline_general.yaml
@@ -89,11 +89,11 @@ INPUT:
     PERCENTAGES: (0.25, 0.25, 0.25, 0.25)
 
 DATALOADER:
-  NUM_WORKERS: 32
+  NUM_WORKERS: 64
   FILTER_EMPTY_ANNOTATIONS: False
 
 SOLVER:
-  IMS_PER_BATCH: 16
+  IMS_PER_BATCH: 32
   CHECKPOINT_PERIOD: 25000
   BASE_LR: 0.0002
   GAMMA: 0.1

diff --git a/configs/segmentation/baseline/baseline_no_augments.yaml b/configs/segmentation/baseline/baseline_no_augments.yaml
@@ -98,7 +98,7 @@ SOLVER:
   BASE_LR: 0.0002
   GAMMA: 0.1
   STEPS: () #(80000, 120000, 160000)
-  MAX_ITER: 750000
+  MAX_ITER: 250000
 
 MODEL:
   MODE: baseline