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):
69
69
class AquaFineTuningParams (DataClassSerializable ):
70
70
epochs : int = None
71
71
learning_rate : float = None
72
+ sample_packing : str = "True"
72
73
73
74
74
75
@dataclass (repr = False )
@@ -562,7 +563,7 @@ def _build_fine_tuning_runtime(
562
563
}
563
564
),
564
565
"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 } "
566
567
+ (f"{ finetuning_params } " if finetuning_params else "" )
567
568
),
568
569
"CONDA_BUCKET_NS" : CONDA_BUCKET_NS ,
You can’t perform that action at this time.
0 commit comments