Skip to content

Commit 08b3dfe

Browse files
Addressing review comment
1 parent 0c6b422 commit 08b3dfe

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ads/aqua/finetuning/finetuning.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -532,7 +532,9 @@ def _build_oci_launch_cmd(
532532
early_stopping_threshold: float = 0.0
533533
) -> str:
534534
"""Builds the oci launch cmd for fine tuning container runtime."""
535-
oci_launch_cmd = f"--training_data {dataset_path} --output_dir {report_path} --val_set_size {val_set_size} --early_stopping_patience {early_stopping_patience} --early_stopping_threshold {early_stopping_threshold}"
535+
oci_launch_cmd = f"--training_data {dataset_path} --output_dir {report_path} --val_set_size {val_set_size}"
536+
if early_stopping_patience:
537+
oci_launch_cmd += f" --early_stopping_patience {early_stopping_patience} --early_stopping_threshold {early_stopping_threshold}"
536538
for key, value in asdict(parameters).items():
537539
if value is not None:
538540
if key == "batch_size":

0 commit comments

Comments
 (0)