summaryrefslogtreecommitdiff
path: root/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml
diff options
context:
space:
mode:
Diffstat (limited to 'src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml')
-rw-r--r--src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml42
1 files changed, 42 insertions, 0 deletions
diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml b/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml
new file mode 100644
index 0000000..0b9b10e
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml
@@ -0,0 +1,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