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. --- .../0721_191412/config.yml | 42 +++++++++++++++++++++ .../0721_191412/model/best.pt | Bin 0 -> 1135058 bytes .../0721_191412/model/last.pt | Bin 0 -> 1135058 bytes 3 files changed, 42 insertions(+) create mode 100644 src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml create mode 100644 src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/model/best.pt create mode 100644 src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/model/last.pt (limited to 'src/training/experiments/CharacterModel_Emnist_MLP/0721_191412') diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml b/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml new file mode 100644 index 0000000..0b9b10e --- /dev/null +++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/config.yml @@ -0,0 +1,42 @@ +criterion: CrossEntropyLoss +criterion_args: + ignore_index: -100 + reduction: mean + weight: null +data_loader_args: + batch_size: 256 + cuda: true + num_workers: 1 + 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: null +metrics: +- accuracy +model: CharacterModel +network: MLP +network_args: + input_size: 784 + num_layers: 3 + output_size: 62 +optimizer: RMSprop +optimizer_args: + alpha: 0.9 + centered: false + eps: 1.0e-07 + lr: 0.001 + momentum: 0 + weight_decay: 0 +train_args: + batch_size: 256 + epochs: 16 + val_metric: accuracy diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/model/best.pt b/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/model/best.pt new file mode 100644 index 0000000..2fb0195 Binary files /dev/null and b/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/model/best.pt differ diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/model/last.pt b/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/model/last.pt new file mode 100644 index 0000000..2fb0195 Binary files /dev/null and b/src/training/experiments/CharacterModel_Emnist_MLP/0721_191412/model/last.pt differ -- cgit v1.2.3-70-g09d2