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.2
56
- model.pool_size:
57
- type: categorical
58
- choices: [1, 2, 3]
59
  model.conv1_size:
60
  type: categorical
61
- choices: [3, 5, 7, 9]
62
  model.conv1_channels:
63
  type: categorical
64
- choices: [1, 3, 6, 9]
65
  model.conv2_size:
66
  type: categorical
67
- choices: [3, 5, 7, 9]
68
  model.conv2_channels:
69
  type: categorical
70
- choices: [1, 3, 6, 9]
71
  model.lin1_size:
72
  type: categorical
73
- choices: [16, 32, 64, 96, 128]
74
  model.lin2_size:
75
  type: categorical
76
- choices: [16, 32, 64, 96, 128]
 
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]