summaryrefslogtreecommitdiff
path: root/src/training/sweep_emnist.yml
blob: 48d726155e0efac5c8bbc06a5e3c32af1b2ed8a9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
program: training/run_sweep.py
method: bayes
metric:
  name: val_loss
  goal: minimize
parameters:
  dataset:
    value: EmnistDataset
  model:
    value: CharacterModel
  network:
    value: MLP
  network_args.hidden_size:
    values: [128, 256]
  network_args.dropout_rate:
    values: [0.2, 0.4]
  network_args.num_layers:
    values: [3, 6]
  optimizer_args.lr:
    values: [1.e-1, 1.e-5]
  lr_scheduler_args.max_lr:
    values: [1.0e-1, 1.0e-5]
  train_args.batch_size:
    values: [64, 128]
  train_args.epochs:
    value: 5