File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -69,6 +69,7 @@ class FineTuneCustomMetadata(Enum):
6969class 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 ,
You can’t perform that action at this time.
0 commit comments