rename to calibration_batch_size

kylesayrs · kylesayrs · commit fe67d7e75a26 · 2025-02-05T17:08:26.000Z
Signed-off-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/src/llmcompressor/transformers/finetune/runner.py b/src/llmcompressor/transformers/finetune/runner.py
@@ -144,7 +144,7 @@ def one_shot(self, stage: Optional[str] = None):
             calib_data = format_calibration_data(
                 tokenized_dataset=self.get_dataset_split("calibration"),
                 num_calibration_samples=self._data_args.num_calibration_samples,
-                batch_size=self._training_args.oneshot_batch_size,
+                batch_size=self._training_args.calibration_batch_size,
                 do_shuffle=self._data_args.shuffle_calibration_samples,
                 collate_fn=self._data_args.data_collator,
                 processor=self.processor,
diff --git a/src/llmcompressor/transformers/finetune/training_args.py b/src/llmcompressor/transformers/finetune/training_args.py
@@ -32,7 +32,7 @@ class TrainingArguments(HFTrainingArgs):
             )
         },
     )
-    oneshot_batch_size: int = field(
+    calibration_batch_size: int = field(
         default=1,
         metadata={
             "help": "The batch size per GPU/XPU/TPU/MPS/NPU core/CPU for oneshot"

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@ class TrainingArguments(HFTrainingArgs):`
`32`	`32`	`)`
`33`	`33`	`},`
`34`	`34`	`)`
`35`		`- oneshot_batch_size: int = field(`
	`35`	`+ calibration_batch_size: int = field(`
`36`	`36`	`default=1,`
`37`	`37`	`metadata={`
`38`	`38`	`"help": "The batch size per GPU/XPU/TPU/MPS/NPU core/CPU for oneshot"`