summaryrefslogtreecommitdiff
path: root/src/training/experiments/sample.yml
diff options
context:
space:
mode:
Diffstat (limited to 'src/training/experiments/sample.yml')
-rw-r--r--src/training/experiments/sample.yml43
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