summaryrefslogtreecommitdiff
path: root/src/training/experiments/default_config_emnist.yml
diff options
context:
space:
mode:
authoraktersnurra <gustaf.rydholm@gmail.com>2020-09-14 22:15:47 +0200
committeraktersnurra <gustaf.rydholm@gmail.com>2020-09-14 22:15:47 +0200
commit3b06ef615a8db67a03927576e0c12fbfb2501f5f (patch)
treee1c2b1289971c8480327408de46152481e99b539 /src/training/experiments/default_config_emnist.yml
parent2b63fd952bdc9c7c72edd501cbcdbf3231e98f00 (diff)
Fixed CTC loss.
Diffstat (limited to 'src/training/experiments/default_config_emnist.yml')
-rw-r--r--src/training/experiments/default_config_emnist.yml69
1 files changed, 69 insertions, 0 deletions
diff --git a/src/training/experiments/default_config_emnist.yml b/src/training/experiments/default_config_emnist.yml
new file mode 100644
index 0000000..12a0a9d
--- /dev/null
+++ b/src/training/experiments/default_config_emnist.yml
@@ -0,0 +1,69 @@
+dataset: EmnistDataset
+dataset_args:
+ sample_to_balance: true
+ subsample_fraction: 0.33
+ transform: null
+ target_transform: null
+ seed: 4711
+
+data_loader_args:
+ splits: [train, val]
+ shuffle: true
+ num_workers: 8
+ cuda: true
+
+model: CharacterModel
+metrics: [accuracy]
+
+network_args:
+ in_channels: 1
+ num_classes: 80
+ depths: [2]
+ block_sizes: [256]
+
+train_args:
+ batch_size: 256
+ epochs: 5
+
+criterion: CrossEntropyLoss
+criterion_args:
+ weight: null
+ ignore_index: -100
+ reduction: mean
+
+optimizer: AdamW
+optimizer_args:
+ lr: 1.e-03
+ betas: [0.9, 0.999]
+ eps: 1.e-08
+ # weight_decay: 5.e-4
+ amsgrad: false
+
+lr_scheduler: OneCycleLR
+lr_scheduler_args:
+ max_lr: 1.e-03
+ epochs: 5
+ anneal_strategy: linear
+
+
+callbacks: [Checkpoint, ProgressBar, EarlyStopping, WandbCallback, WandbImageLogger, OneCycleLR]
+callback_args:
+ Checkpoint:
+ monitor: val_accuracy
+ ProgressBar:
+ epochs: 5
+ log_batch_frequency: 100
+ EarlyStopping:
+ monitor: val_loss
+ min_delta: 0.0
+ patience: 3
+ mode: min
+ WandbCallback:
+ log_batch_frequency: 10
+ WandbImageLogger:
+ num_examples: 4
+ OneCycleLR:
+ null
+verbosity: 1 # 0, 1, 2
+resume_experiment: null
+validation_metric: val_accuracy