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]
|