We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2d50595 commit 8d41c73Copy full SHA for 8d41c73
src/templates/template-vision-classification/config.yaml
@@ -1,3 +1,19 @@
1
-#::= from_template_common ::#
2
-lr: 0.0001 # learning rate
3
-model: resnet18 # model name
+seed: 777 # random seed
+data_path: ./ # input data path
+batch_size: 400 # train batch size
4
+eval_batch_size: 400 # evaluation batch size
5
+num_workers: 10 # number of subprocesses to use for data loading
6
+max_epochs: 24 # number of maximum epochs
7
+use_amp: false # use AMP (Automatic Mixed Precision)
8
+weight_decay: 1e-4
9
+debug: false
10
+train_epoch_length: null
11
+eval_epoch_length: null
12
+filename_prefix: training
13
+n_saved: 2
14
+save_every_iters: 1000
15
+patience: 10
16
+output_dir: ./logs
17
+log_every_iters: 10
18
+lr: 1e-2 # learning rate
19
+model: resnet18 # model name
0 commit comments