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
|
