Skip to content

Commit 2097092

Browse files
committed
Set new hyperparameter values for various strategies after more extensive training
1 parent f56c1fd commit 2097092

File tree

4 files changed

+14
-14
lines changed

4 files changed

+14
-14
lines changed

kernel_tuner/strategies/diff_evo.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,11 @@
1010
from kernel_tuner.strategies.common import CostFunc
1111

1212
_options = dict(
13-
popsize=("population size", 50),
14-
popsize_times_dimensions=("multiply population size with number of dimensions (True/False)", False),
13+
popsize=("population size", 16),
14+
popsize_times_dimensions=("multiply population size with number of dimensions (True/False)", True),
1515
maxiter=("maximum number of generations", int(1e15)), # very large to avoid early stopping (stopping is managed by StopCriterionReached)
16-
F=("mutation factor (differential weight)", 1.3),
17-
CR=("crossover rate", 0.9),
16+
F=("mutation factor (differential weight)", 0.7),
17+
CR=("crossover rate", 0.6),
1818
method=("method", "best1bin"),
1919
constraint_aware=("constraint-aware optimization (True/False)", True),
2020
)

kernel_tuner/strategies/genetic_algorithm.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,10 @@
1010
from kernel_tuner.strategies.common import CostFunc
1111

1212
_options = dict(
13-
popsize=("population size", 20),
14-
maxiter=("maximum number of generations", 150),
13+
popsize=("population size", 26),
14+
maxiter=("maximum number of generations", 90),
1515
method=("crossover method to use, choose any from single_point, two_point, uniform, disruptive_uniform", "single_point"),
16-
mutation_chance=("chance to mutate is 1 in mutation_chance", 5),
16+
mutation_chance=("chance to mutate is 1 in mutation_chance", 55),
1717
constraint_aware=("constraint-aware optimization (True/False)", True),
1818
)
1919

kernel_tuner/strategies/pso.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,11 @@
1111
from kernel_tuner.strategies.common import CostFunc, scale_from_params
1212

1313
_options = dict(
14-
popsize=("Population size", 30),
15-
maxiter=("Maximum number of iterations", 100),
14+
popsize=("Population size", 50),
15+
maxiter=("Maximum number of iterations", 190),
1616
w=("Inertia weight constant", 0.5),
17-
c1=("Cognitive constant", 3.0),
18-
c2=("Social constant", 0.5),
17+
c1=("Cognitive constant", 3.5),
18+
c2=("Social constant", 1.0),
1919
constraint_aware=("constraint-aware optimization (True/False)", True))
2020

2121
def tune(searchspace: Searchspace, runner, tuning_options):

kernel_tuner/strategies/simulated_annealing.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,10 @@
1010
from kernel_tuner.strategies.common import CostFunc
1111

1212

13-
_options = dict(T=("Starting temperature", 0.5),
14-
T_min=("End temperature", 0.001),
13+
_options = dict(T=("Starting temperature", 0.1),
14+
T_min=("End temperature", 0.0001),
1515
alpha=("Alpha parameter", 0.9975),
16-
maxiter=("Number of iterations within each annealing step", 2),
16+
maxiter=("Number of iterations within each annealing step", 1),
1717
constraint_aware=("constraint-aware optimization (True/False)", True))
1818

1919
def tune(searchspace: Searchspace, runner, tuning_options):

0 commit comments

Comments
 (0)