From b01dd74c3a2bd98c9df680cc4c1a7cabe8bf18f2 Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Mon, 17 Apr 2023 15:28:11 -0400 Subject: [PATCH] [fix] remove image_size from call to rescale_gradient_accumulation (#202) (#204) #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