blob: fb7573658113d449909cda76912afe18fb985a58 (
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
criterion: CrossEntropyLoss
criterion_args:
ignore_index: -100
reduction: mean
weight: null
data_loader_args:
batch_size: 256
cuda: true
num_workers: 0
sample_to_balance: true
seed: 4711
shuffle: true
splits:
- train
- val
subsample_fraction: null
target_transform: null
transform: null
dataloader: EmnistDataLoader
device: cuda:0
experiment_group: Sample Experiments
lr_scheduler: OneCycleLR
lr_scheduler_args:
epochs: 16
max_lr: 0.001
steps_per_epoch: 1314
metrics:
- accuracy
model: CharacterModel
network: MLP
network_args:
input_size: 784
num_layers: 3
output_size: 62
optimizer: AdamW
optimizer_args:
amsgrad: false
betas:
- 0.9
- 0.999
eps: 1.0e-08
lr: 0.01
weight_decay: 5.0e-05
train_args:
batch_size: 256
epochs: 16
val_metric: accuracy
|
