diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-07-14 14:54:11 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-07-14 14:55:00 +0800 |
| commit | cc807c751e2c14ef9a88e5c5be00b4eb082e705b (patch) | |
| tree | 04af8607e2906df68d3e77edbd2658353fb044a0 /supervised/config.yaml | |
| parent | b9d83e80b946437bb8dc0b586488fa756f52d732 (diff) | |
Refactor baseline with trainer
Diffstat (limited to 'supervised/config.yaml')
| -rw-r--r-- | supervised/config.yaml | 38 |
1 files changed, 16 insertions, 22 deletions
diff --git a/supervised/config.yaml b/supervised/config.yaml index 4fb37fb..91d6dc0 100644 --- a/supervised/config.yaml +++ b/supervised/config.yaml @@ -1,31 +1,25 @@ -codename: cifar10-resnet50-256-lars-warmup +codename: cifar10-resnet50-256-adam-linear seed: -1 +num_iters: 1000 +log_dir: logs +checkpoint_dir: checkpoints -dataset_dir: dataset dataset: cifar10 +dataset_dir: dataset 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 + - 0.8 + - 1 +hflip_prob: 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 +num_workers: 2 + +optim: adam +lr: 0.001 +betas: + - 0.9 + - 0.999 weight_decay: 1.0e-06 -log_dir: logs -tensorboard_dir: runs -checkpoint_dir: checkpoints
\ No newline at end of file +sched: linear
\ No newline at end of file |