diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-03-17 18:55:53 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-03-17 18:55:53 +0800 |
| commit | 1de6b8270c34390f2000cb52a124170dbfda6dc3 (patch) | |
| tree | 3ec746fab834b17f229d3b8e5d7037a4ef4c2ea4 /supervised/config.yaml | |
| parent | 48e0405bfbffa63d9e2a610c1cf982892a861e4a (diff) | |
Add YAML parser support
Diffstat (limited to 'supervised/config.yaml')
| -rw-r--r-- | supervised/config.yaml | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/supervised/config.yaml b/supervised/config.yaml new file mode 100644 index 0000000..2279b3c --- /dev/null +++ b/supervised/config.yaml @@ -0,0 +1,27 @@ +codename: 'cifar10-resnet50-256-lars-warmup' +seed: -1 + +dataset_dir: 'dataset' +dataset: 'cifar10' +crop_size: 32 +crop_scale_range: '0.8-1' +hflip_p: 0.5 +distort_s: 0.5 +gaussian_ker_scale: 10 +gaussian_sigma_range: '0.1-2' +gaussian_p: 0.5 + +batch_size: 256 +restore_epoch: 0 +n_epochs: 1000 +warmup_epochs: 10 +n_workers: 2 +optim: 'lars' +sched: 'warmup-anneal' +lr: 1. +momentum: 0.9 +weight_decay: 1.0e-06 + +log_dir: 'logs' +tensorboard_dir: 'runs' +checkpoint_dir: 'checkpoints'
\ No newline at end of file |