diff --git a/rfdetr/config.py b/rfdetr/config.py
index 8999cdf..4ba9b74 100644
--- a/rfdetr/config.py
+++ b/rfdetr/config.py
@@ -162,6 +162,7 @@ class TrainConfig(BaseModel):
 
 
 class SegmentationTrainConfig(TrainConfig):
+    num_select: int = 200
     mask_point_sample_ratio: int = 16
     mask_ce_loss_coef: float = 5.0
     mask_dice_loss_coef: float = 5.0