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_231455/config.yml                         |  48 +++++++++++++++++++++
 .../0721_231455/model/best.pt                      | Bin 0 -> 14483400 bytes
 .../0721_231455/model/last.pt                      | Bin 0 -> 14483400 bytes
 3 files changed, 48 insertions(+)
 create mode 100644 src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/config.yml
 create mode 100644 src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/model/best.pt
 create mode 100644 src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/model/last.pt

(limited to 'src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455')

diff --git a/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/config.yml b/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/config.yml
new file mode 100644
index 0000000..2595325
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/config.yml
@@ -0,0 +1,48 @@
+criterion: CrossEntropyLoss
+criterion_args:
+  ignore_index: -100
+  reduction: mean
+  weight: null
+data_loader_args:
+  batch_size: 256
+  cuda: true
+  num_workers: 8
+  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: LeNet
+network_args:
+  input_size:
+  - 28
+  - 28
+  output_size: 62
+optimizer: AdamW
+optimizer_args:
+  amsgrad: false
+  betas:
+  - 0.9
+  - 0.999
+  eps: 1.0e-08
+  lr: 0.01
+  weight_decay: 0
+train_args:
+  batch_size: 256
+  epochs: 16
+  val_metric: accuracy
diff --git a/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/model/best.pt b/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/model/best.pt
new file mode 100644
index 0000000..6d78bad
Binary files /dev/null and b/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/model/best.pt differ
diff --git a/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/model/last.pt b/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/model/last.pt
new file mode 100644
index 0000000..6d78bad
Binary files /dev/null and b/src/training/experiments/CharacterModel_Emnist_LeNet/0721_231455/model/last.pt differ
-- 
cgit v1.2.3-70-g09d2