From f473456c19558aaf8552df97a51d4e18cc69dfa8 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Wed, 22 Jul 2020 23:18:08 +0200
Subject: Working training loop and testing of trained CharacterModel.

---
 .../0721_145028/config.yml                         | 47 ++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 src/training/experiments/CharacterModel_Emnist_MLP/0721_145028/config.yml

(limited to 'src/training/experiments/CharacterModel_Emnist_MLP/0721_145028')

diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_145028/config.yml b/src/training/experiments/CharacterModel_Emnist_MLP/0721_145028/config.yml
new file mode 100644
index 0000000..fb75736
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_145028/config.yml
@@ -0,0 +1,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
-- 
cgit v1.2.3-70-g09d2