diff --git a/configs/segmentation/baseline/baseline_general.yaml b/configs/segmentation/baseline/baseline_general.yaml
index 1ebb832..89b81f5 100644
--- 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
index 1a12fd3..94c2bb5 100644
--- 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