diff --git a/osrt/encoder.py b/osrt/encoder.py
index 64f61ddb4f1f7ce830605a785bd602b400399b07..a96650e3c43aa6cc4eee6834cbadd47a34aff773 100644
--- a/osrt/encoder.py
+++ b/osrt/encoder.py
@@ -120,9 +120,9 @@ class FeatureMasking(nn.Module):
                  pred_iou_thresh=0.88,
                  points_per_batch=64,
                  min_mask_region_area=0,
-                 num_slots=10, 
+                 num_slots=32, 
                  slot_dim=1536, 
-                 slot_iters=1,  
+                 slot_iters=3,  
                  sam_model="default", 
                  sam_path="sam_vit_h_4b8939.pth",
                  randomize_initial_slots=False):