From 568569c764ffdd73cd660434df50d30d26203f63 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 17 Mar 2022 19:12:50 +0800 Subject: Log parameters and use list YAML syntax for tuples --- supervised/config.yaml | 24 ++++++++++++++---------- 1 file changed, 14 insertions(+), 10 deletions(-) (limited to 'supervised/config.yaml') diff --git a/supervised/config.yaml b/supervised/config.yaml index 2279b3c..4fb37fb 100644 --- a/supervised/config.yaml +++ b/supervised/config.yaml @@ -1,14 +1,18 @@ -codename: 'cifar10-resnet50-256-lars-warmup' +codename: cifar10-resnet50-256-lars-warmup seed: -1 -dataset_dir: 'dataset' -dataset: 'cifar10' +dataset_dir: dataset +dataset: cifar10 crop_size: 32 -crop_scale_range: '0.8-1' +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_sigma_range: +- 0.1 +- 2 gaussian_p: 0.5 batch_size: 256 @@ -16,12 +20,12 @@ restore_epoch: 0 n_epochs: 1000 warmup_epochs: 10 n_workers: 2 -optim: 'lars' -sched: 'warmup-anneal' +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 +log_dir: logs +tensorboard_dir: runs +checkpoint_dir: checkpoints \ No newline at end of file -- cgit v1.2.3