From cc807c751e2c14ef9a88e5c5be00b4eb082e705b Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 14 Jul 2022 14:54:11 +0800 Subject: Refactor baseline with trainer --- supervised/config.yaml | 38 ++++++++++++++++---------------------- 1 file changed, 16 insertions(+), 22 deletions(-) (limited to 'supervised/config.yaml') 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 -- cgit v1.2.3