Hannes Kuchelmeister
commited on
Commit
·
26b8b68
1
Parent(s):
907dce4
reduce search space for hyperparameter optimisation
Browse files
configs/hparams_search/focusConvMSE_150_hyperparameter_search.yaml
CHANGED
@@ -52,25 +52,22 @@ hydra:
|
|
52 |
model.lr:
|
53 |
type: float
|
54 |
low: 0.0001
|
55 |
-
high: 0.
|
56 |
-
model.pool_size:
|
57 |
-
type: categorical
|
58 |
-
choices: [1, 2, 3]
|
59 |
model.conv1_size:
|
60 |
type: categorical
|
61 |
-
choices: [3, 5, 7
|
62 |
model.conv1_channels:
|
63 |
type: categorical
|
64 |
-
choices: [
|
65 |
model.conv2_size:
|
66 |
type: categorical
|
67 |
-
choices: [3, 5, 7
|
68 |
model.conv2_channels:
|
69 |
type: categorical
|
70 |
-
choices: [
|
71 |
model.lin1_size:
|
72 |
type: categorical
|
73 |
-
choices: [
|
74 |
model.lin2_size:
|
75 |
type: categorical
|
76 |
-
choices: [
|
|
|
52 |
model.lr:
|
53 |
type: float
|
54 |
low: 0.0001
|
55 |
+
high: 0.01
|
|
|
|
|
|
|
56 |
model.conv1_size:
|
57 |
type: categorical
|
58 |
+
choices: [3, 5, 7]
|
59 |
model.conv1_channels:
|
60 |
type: categorical
|
61 |
+
choices: [3, 6, 9]
|
62 |
model.conv2_size:
|
63 |
type: categorical
|
64 |
+
choices: [3, 5, 7]
|
65 |
model.conv2_channels:
|
66 |
type: categorical
|
67 |
+
choices: [6, 11, 16]
|
68 |
model.lin1_size:
|
69 |
type: categorical
|
70 |
+
choices: [32, 72, 128]
|
71 |
model.lin2_size:
|
72 |
type: categorical
|
73 |
+
choices: [32, 72, 128]
|