diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-19 14:02:04 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-19 14:02:04 +0800 |
commit | 26420733f98292639b9addb02e73fd8f12ee82e7 (patch) | |
tree | 971d3e5223be2261562e6ca404378aa34aaee453 /supervised | |
parent | bff36e9337bd9493e95588b8e342431eb31184f6 (diff) |
Add dependencies
- Pytorch Lighting (pytorch-lighting)
- Pytorch image models (timm)
Diffstat (limited to 'supervised')
-rw-r--r-- | supervised/config-resnet-ddp.yaml | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/supervised/config-resnet-ddp.yaml b/supervised/config-resnet-ddp.yaml new file mode 100644 index 0000000..6fc8007 --- /dev/null +++ b/supervised/config-resnet-ddp.yaml @@ -0,0 +1,29 @@ +codename: cifar10-resnet-128-adam-warmup-anneal-ddp +seed: -1 +num_iters: 200 +log_dir: logs +checkpoint_dir: checkpoints + +backbone: resnet +label_smooth: 0 + +dataset: cifar10 +dataset_dir: dataset +crop_size: 32 +crop_scale_range: + - 0.8 + - 1 +hflip_prob: 0.5 + +batch_size: 128 +num_workers: 2 + +optim: adam +lr: 0.001 +betas: + - 0.9 + - 0.999 +weight_decay: 1.0e-06 + +sched: warmup-anneal +warmup_iters: 5
\ No newline at end of file |