Hannes Kuchelmeister
commited on
Commit
•
6c09f54
1
Parent(s):
1a50bbb
reduce search space for optuna experiments
Browse files
configs/hparams_search/focus150_optuna.yaml
CHANGED
@@ -48,17 +48,17 @@ hydra:
|
|
48 |
search_space:
|
49 |
datamodule.batch_size:
|
50 |
type: categorical
|
51 |
-
choices: [
|
52 |
model.lr:
|
53 |
type: float
|
54 |
low: 0.0001
|
55 |
high: 0.2
|
56 |
model.lin1_size:
|
57 |
type: categorical
|
58 |
-
choices: [
|
59 |
model.lin2_size:
|
60 |
type: categorical
|
61 |
-
choices: [
|
62 |
model.lin3_size:
|
63 |
type: categorical
|
64 |
-
choices: [
|
|
|
48 |
search_space:
|
49 |
datamodule.batch_size:
|
50 |
type: categorical
|
51 |
+
choices: [64, 128]
|
52 |
model.lr:
|
53 |
type: float
|
54 |
low: 0.0001
|
55 |
high: 0.2
|
56 |
model.lin1_size:
|
57 |
type: categorical
|
58 |
+
choices: [64, 128, 256, 512, 1024]
|
59 |
model.lin2_size:
|
60 |
type: categorical
|
61 |
+
choices: [64, 128, 256, 512, 1024]
|
62 |
model.lin3_size:
|
63 |
type: categorical
|
64 |
+
choices: [64, 128, 256, 512, 1024]
|
configs/hparams_search/focus_optuna.yaml
CHANGED
@@ -46,17 +46,17 @@ hydra:
|
|
46 |
search_space:
|
47 |
datamodule.batch_size:
|
48 |
type: categorical
|
49 |
-
choices: [
|
50 |
model.lr:
|
51 |
type: float
|
52 |
low: 0.0001
|
53 |
high: 0.2
|
54 |
model.lin1_size:
|
55 |
type: categorical
|
56 |
-
choices: [
|
57 |
model.lin2_size:
|
58 |
type: categorical
|
59 |
-
choices: [
|
60 |
model.lin3_size:
|
61 |
type: categorical
|
62 |
-
choices: [
|
|
|
46 |
search_space:
|
47 |
datamodule.batch_size:
|
48 |
type: categorical
|
49 |
+
choices: [64, 128]
|
50 |
model.lr:
|
51 |
type: float
|
52 |
low: 0.0001
|
53 |
high: 0.2
|
54 |
model.lin1_size:
|
55 |
type: categorical
|
56 |
+
choices: [64, 128, 256, 512, 1024]
|
57 |
model.lin2_size:
|
58 |
type: categorical
|
59 |
+
choices: [64, 128, 256, 512, 1024]
|
60 |
model.lin3_size:
|
61 |
type: categorical
|
62 |
+
choices: [64, 128, 256, 512, 1024]
|