Skip to content

Commit 7f3a4a3

Browse files
committed
Updated hyperparameters to newly tuned defaults
1 parent 5f3b6fc commit 7f3a4a3

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

kernel_tuner/strategies/genetic_algorithm.py

Lines changed: 4 additions & 4 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", 30),
14-
maxiter=("maximum number of generations", 30),
15-
method=("crossover method to use, choose any from single_point, two_point, uniform, disruptive_uniform", "uniform"),
16-
mutation_chance=("chance to mutate is 1 in mutation_chance", 20),
13+
popsize=("population size", 20),
14+
maxiter=("maximum number of generations", 150),
15+
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),
1717
constraint_aware=("constraint-aware optimization (True/False)", True),
1818
)
1919

kernel_tuner/strategies/pso.py

Lines changed: 3 additions & 3 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", 20),
15-
maxiter=("Maximum number of iterations", 150),
14+
popsize=("Population size", 30),
15+
maxiter=("Maximum number of iterations", 100),
1616
w=("Inertia weight constant", 0.5),
1717
c1=("Cognitive constant", 3.0),
18-
c2=("Social constant", 1.5),
18+
c2=("Social constant", 0.5),
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: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212

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

0 commit comments

Comments
 (0)