File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff 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" :
You can’t perform that action at this time.
0 commit comments