add early stopping for hyperopt

This commit is contained in:
viotemp1
2025-05-10 19:36:48 +03:00
parent 2a0dd4cf55
commit 442a1ba50d
6 changed files with 34 additions and 0 deletions

View File

@@ -262,6 +262,14 @@ AVAILABLE_CLI_OPTIONS = {
metavar="INT",
default=constants.HYPEROPT_EPOCH,
),
"early_stop": Arg(
"-es",
"--early-stop",
help="Early stop hyperopt if no improvement after (default: %(default)d) epochs.",
type=check_int_positive,
metavar="INT",
default=0, # 0 to disable by default
),
"spaces": Arg(
"--spaces",
help="Specify which parameters to hyperopt. Space-separated list.",