diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2020-07-22 23:18:08 +0200 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2020-07-22 23:18:08 +0200 |
| commit | f473456c19558aaf8552df97a51d4e18cc69dfa8 (patch) | |
| tree | 0d35ce2410ff623ba5fb433d616d95b67ecf7a98 /src/training/experiments/CharacterModel_Emnist_MLP/0721_151408/config.yml | |
| parent | ad3bd52530f4800d4fb05dfef3354921f95513af (diff) | |
Working training loop and testing of trained CharacterModel.
Diffstat (limited to 'src/training/experiments/CharacterModel_Emnist_MLP/0721_151408/config.yml')
| -rw-r--r-- | src/training/experiments/CharacterModel_Emnist_MLP/0721_151408/config.yml | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/src/training/experiments/CharacterModel_Emnist_MLP/0721_151408/config.yml b/src/training/experiments/CharacterModel_Emnist_MLP/0721_151408/config.yml new file mode 100644 index 0000000..fb75736 --- /dev/null +++ b/src/training/experiments/CharacterModel_Emnist_MLP/0721_151408/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 |