Working training loop and testing of trained CharacterModel.

1 files changed, 43 insertions, 0 deletions

diff --git a/src/training/experiments/sample.yml b/src/training/experiments/sample.yml
new file mode 100644
index 0000000..0ed560d
--- /dev/null
+++ b/src/training/experiments/sample.yml
@@ -0,0 +1,43 @@
+experiment_group: Sample Experiments
+experiments:
+    - dataloader: EmnistDataLoader
+      model: CharacterModel
+      metrics: [accuracy]
+      network: MLP
+      network_args:
+        input_shape: 784
+        num_layers: 2
+      train_args:
+        batch_size: 256
+        epochs: 16
+      criterion: CrossEntropyLoss
+      criterion_args:
+        weight: null
+        ignore_index: -100
+        reduction: mean
+      optimizer: AdamW
+      optimizer_args:
+        lr: 3.e-4
+        betas: [0.9, 0.999]
+        eps: 1.e-08
+        weight_decay: 0
+        amsgrad: false
+      lr_scheduler: OneCycleLR
+      lr_scheduler_args:
+        max_lr: 3.e-5
+        epochs: 16
+    # - dataloader: EmnistDataLoader
+    #   model: CharacterModel
+    #   network: MLP
+    #   network_args:
+    #     input_shape: 784
+    #     num_layers: 4
+    #   train_args:
+    #     batch_size: 256
+    # - dataloader: EmnistDataLoader
+    #   model: CharacterModel
+    #   network: LeNet
+    #   network_args:
+    #     input_shape: [28, 28]
+    #   train_args:
+    #     batch_size: 256