diff options
Diffstat (limited to 'src/training/experiments/sample.yml')
| -rw-r--r-- | src/training/experiments/sample.yml | 43 |
1 files changed, 0 insertions, 43 deletions
diff --git a/src/training/experiments/sample.yml b/src/training/experiments/sample.yml deleted file mode 100644 index 0ed560d..0000000 --- a/src/training/experiments/sample.yml +++ /dev/null @@ -1,43 +0,0 @@ -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 |