blob: 0b9b10e7e55fc30a7d511ba4ae62b86679377ac5 (
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
|
criterion: CrossEntropyLoss
criterion_args:
ignore_index: -100
reduction: mean
weight: null
data_loader_args:
batch_size: 256
cuda: true
num_workers: 1
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: null
metrics:
- accuracy
model: CharacterModel
network: MLP
network_args:
input_size: 784
num_layers: 3
output_size: 62
optimizer: RMSprop
optimizer_args:
alpha: 0.9
centered: false
eps: 1.0e-07
lr: 0.001
momentum: 0
weight_decay: 0
train_args:
batch_size: 256
epochs: 16
val_metric: accuracy
|
