From f473456c19558aaf8552df97a51d4e18cc69dfa8 Mon Sep 17 00:00:00 2001 From: aktersnurra Date: Wed, 22 Jul 2020 23:18:08 +0200 Subject: Working training loop and testing of trained CharacterModel. --- src/training/experiments/sample.yml | 43 +++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 src/training/experiments/sample.yml (limited to 'src/training/experiments/sample.yml') 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 -- cgit v1.2.3-70-g09d2