Working training loop and testing of trained CharacterModel.

3 files changed, 47 insertions, 0 deletions

diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/config.yml b/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/config.yml
new file mode 100644
index 0000000..829297d
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/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.0003
+  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.0006
+  weight_decay: 5.0e-05
+train_args:
+  batch_size: 256
+  epochs: 16
+  val_metric: accuracy
diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/model/best.pt b/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/model/best.pt
new file mode 100644
index 0000000..d0db78b
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/model/best.pt
diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/model/last.pt b/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/model/last.pt
new file mode 100644
index 0000000..d0db78b
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_141213/model/last.pt