[fix] remove image_size from call to rescale_gradient_accumulation (#202) (#203)

bfineran · web-flow · commit 9bad2683cfee · 2023-04-17T15:28:06.000-04:00
#200 included a side change to remove the now unused `image_size` arg from the function signature but did not update the call site. this PR fixes that. **test_plan:** Reproduced reported error with a sample quantization recipe and verified this PR resolves it
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -314,7 +314,6 @@ def _create_dataloaders():`
`314`	`314`	`new_batch_size, new_accumulate = sparsification_manager.rescale_gradient_accumulation(`
`315`	`315`	`batch_size=batch_size,`
`316`	`316`	`accumulate=accumulate,`
`317`		`- image_size=imgsz`
`318`	`317`	`)`
`319`	`318`	`if new_batch_size != batch_size:`
`320`	`319`	`batch_size = new_batch_size`