summaryrefslogtreecommitdiff
path: root/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135
diff options
context:
space:
mode:
Diffstat (limited to 'src/training/experiments/CharacterModel_Emnist_MLP/0721_151135')
-rw-r--r--src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/config.yml47
-rw-r--r--src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/model/best.ptbin0 -> 1901268 bytes
-rw-r--r--src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/model/last.ptbin0 -> 1901268 bytes
3 files changed, 47 insertions, 0 deletions
diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/config.yml b/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/config.yml
new file mode 100644
index 0000000..fb75736
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/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.001
+ 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.01
+ 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_151135/model/best.pt b/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/model/best.pt
new file mode 100644
index 0000000..f833a89
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/model/best.pt
Binary files differ
diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/model/last.pt b/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/model/last.pt
new file mode 100644
index 0000000..f833a89
--- /dev/null
+++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_151135/model/last.pt
Binary files differ