From 4b04d4c077c75d1830270d48645a51335e4bee7a Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Mon, 17 Apr 2023 15:23:03 -0400 Subject: [PATCH] [fix] remove image_size from call to rescale_gradient_accumulation (#202) #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 --- train.py | 1 - 1 file changed, 1 deletion(-) diff --git a/train.py b/train.py index 83e8b6f8e6fe..57844fbb9c78 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