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: [32, 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: [32, 64, 128, 256, 512, 1024, 2048]
59
  model.lin2_size:
60
  type: categorical
61
- choices: [32, 64, 128, 256, 512, 1024, 2048]
62
  model.lin3_size:
63
  type: categorical
64
- choices: [32, 64, 128, 256, 512, 1024, 2048]
 
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: [32, 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: [32, 64, 128, 256, 512, 1024, 2048]
57
  model.lin2_size:
58
  type: categorical
59
- choices: [32, 64, 128, 256, 512, 1024, 2048]
60
  model.lin3_size:
61
  type: categorical
62
- choices: [32, 64, 128, 256, 512, 1024, 2048]
 
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]