diff --git a/configs/cityscapes/e2e_mask_rcnn_R_50_FPN_1x_binarymask.yaml b/configs/cityscapes/e2e_mask_rcnn_R_50_FPN_1x_binarymask.yaml
index fd4501da6..32e8ec014 100644
--- a/configs/cityscapes/e2e_mask_rcnn_R_50_FPN_1x_binarymask.yaml
+++ b/configs/cityscapes/e2e_mask_rcnn_R_50_FPN_1x_binarymask.yaml
@@ -20,7 +20,7 @@ MODEL:
     POOLER_SAMPLING_RATIO: 2
     FEATURE_EXTRACTOR: "FPN2MLPFeatureExtractor"
     PREDICTOR: "FPNPredictor"
-    NUM_CLASSES: 9
+    NUM_CLASSES: 11
   ROI_MASK_HEAD:
     POOLER_SCALES: (0.25, 0.125, 0.0625, 0.03125)
     FEATURE_EXTRACTOR: "MaskRCNNFPNFeatureExtractor"
@@ -30,12 +30,18 @@ MODEL:
     RESOLUTION: 28
     SHARE_BOX_FEATURE_EXTRACTOR: False
   MASK_ON: True
+INPUT:
+  MIN_SIZE_TRAIN: (800, 832, 864, 896, 928, 960, 992, 1024, 1024)
+  MAX_SIZE_TRAIN: 2048
+  MIN_SIZE_TEST: 1024
+  MAX_SIZE_TEST: 2048
 DATASETS:
   TRAIN: ("cityscapes_mask_instance_train",)
   TEST: ("cityscapes_mask_instance_val",)
 DATALOADER:
   SIZE_DIVISIBILITY: 32
 SOLVER:
+  IMS_PER_BATCH: 8
   BASE_LR: 0.01
   WEIGHT_DECAY: 0.0001
   STEPS: (18000,)