summaryrefslogtreecommitdiff
path: root/src/training/experiments/sample_experiment.yml
diff options
context:
space:
mode:
authoraktersnurra <gustaf.rydholm@gmail.com>2020-09-20 00:14:27 +0200
committeraktersnurra <gustaf.rydholm@gmail.com>2020-09-20 00:14:27 +0200
commite181195a699d7fa237f256d90ab4dedffc03d405 (patch)
tree6d8d50731a7267c56f7bf3ed5ecec3990c0e55a5 /src/training/experiments/sample_experiment.yml
parent3b06ef615a8db67a03927576e0c12fbfb2501f5f (diff)
Minor bug fixes etc.
Diffstat (limited to 'src/training/experiments/sample_experiment.yml')
-rw-r--r--src/training/experiments/sample_experiment.yml13
1 files changed, 5 insertions, 8 deletions
diff --git a/src/training/experiments/sample_experiment.yml b/src/training/experiments/sample_experiment.yml
index 17e220e..8664a15 100644
--- a/src/training/experiments/sample_experiment.yml
+++ b/src/training/experiments/sample_experiment.yml
@@ -2,7 +2,7 @@ experiment_group: Sample Experiments
experiments:
- train_args:
batch_size: 256
- max_epochs: 32
+ max_epochs: &max_epochs 32
dataset:
type: EmnistDataset
args:
@@ -66,16 +66,17 @@ experiments:
# type: OneCycleLR
# args:
# max_lr: 1.e-03
- # epochs: null
+ # epochs: *max_epochs
# anneal_strategy: linear
lr_scheduler:
type: CosineAnnealingLR
args:
- T_max: null
+ T_max: *max_epochs
+ interval: epoch
swa_args:
start: 2
lr: 5.e-2
- callbacks: [Checkpoint, ProgressBar, WandbCallback, WandbImageLogger, EarlyStopping, SWA] # OneCycleLR]
+ callbacks: [Checkpoint, ProgressBar, WandbCallback, WandbImageLogger, EarlyStopping]
callback_args:
Checkpoint:
monitor: val_accuracy
@@ -92,10 +93,6 @@ experiments:
WandbImageLogger:
num_examples: 4
use_transpose: true
- # OneCycleLR:
- # null
- SWA:
- null
verbosity: 0 # 0, 1, 2
resume_experiment: null
test: true