Skip to content

Commit 2f3c48a

Browse files
authored
Ability to disable packing of dataset in CLI (#804)
1 parent 8ca48d1 commit 2f3c48a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ads/aqua/finetune.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ class FineTuneCustomMetadata(Enum):
6969
class AquaFineTuningParams(DataClassSerializable):
7070
epochs: int = None
7171
learning_rate: float = None
72+
sample_packing: str = "True"
7273

7374

7475
@dataclass(repr=False)
@@ -562,7 +563,7 @@ def _build_fine_tuning_runtime(
562563
}
563564
),
564565
"OCI__LAUNCH_CMD": (
565-
f"--micro_batch_size {batch_size} --num_epochs {parameters.epochs} --learning_rate {parameters.learning_rate} --training_data {dataset_path} --output_dir {report_path} --val_set_size {val_set_size} "
566+
f"--micro_batch_size {batch_size} --num_epochs {parameters.epochs} --learning_rate {parameters.learning_rate} --training_data {dataset_path} --output_dir {report_path} --val_set_size {val_set_size} --sample_packing {parameters.sample_packing} "
566567
+ (f"{finetuning_params}" if finetuning_params else "")
567568
),
568569
"CONDA_BUCKET_NS": CONDA_BUCKET_NS,

0 commit comments

Comments
 (0)