diff --git a/train.py b/train.py
index 83e8b6f8e6f..57844fbb9c7 100644
--- a/train.py
+++ b/train.py
@@ -314,7 +314,6 @@ def _create_dataloaders():
                 new_batch_size, new_accumulate = sparsification_manager.rescale_gradient_accumulation(
                     batch_size=batch_size, 
                     accumulate=accumulate, 
-                    image_size=imgsz
                 )
                 if new_batch_size != batch_size:
                     batch_size = new_batch_size