From 8d41c73394e47664a9a2e13e465b545977a26a87 Mon Sep 17 00:00:00 2001 From: Venkat Ramnan Kalyanakumar Date: Mon, 18 Dec 2023 22:44:26 -0800 Subject: [PATCH] feat: config for matching ignite capapbility --- .../config.yaml | 22 ++++++++++++++++--- 1 file changed, 19 insertions(+), 3 deletions(-) diff --git a/src/templates/template-vision-classification/config.yaml b/src/templates/template-vision-classification/config.yaml index da24c423..9f0fba9f 100644 --- a/src/templates/template-vision-classification/config.yaml +++ b/src/templates/template-vision-classification/config.yaml @@ -1,3 +1,19 @@ -#::= from_template_common ::# -lr: 0.0001 # learning rate -model: resnet18 # model name +seed: 777 # random seed +data_path: ./ # input data path +batch_size: 400 # train batch size +eval_batch_size: 400 # evaluation batch size +num_workers: 10 # number of subprocesses to use for data loading +max_epochs: 24 # number of maximum epochs +use_amp: false # use AMP (Automatic Mixed Precision) +weight_decay: 1e-4 +debug: false +train_epoch_length: null +eval_epoch_length: null +filename_prefix: training +n_saved: 2 +save_every_iters: 1000 +patience: 10 +output_dir: ./logs +log_every_iters: 10 +lr: 1e-2 # learning rate +model: resnet18 # model name \ No newline at end of file