Merge branch 'main' into navcode_fix

Author: guptaaryan16

.github/workflows/release.yml

@@ -1,11 +1,11 @@
 #::= from_template_common ::#
-model: bert-base-uncased
-model_dir: /tmp/model
-tokenizer_dir: /tmp/tokenizer
-num_classes: 1
-drop_out: .3
-n_fc: 768
-weight_decay: 0.01
-num_warmup_epochs: 0
-max_length: 256
+model: bert-base-uncased # model for text classification
+model_dir: /tmp/model # path of model directory
+tokenizer_dir: /tmp/tokenizer # path of directory containing tokenizer
+num_classes: 1 # number of classes
+drop_out: .3 # drop out rate
+n_fc: 768 # number of fully-connected layers
+weight_decay: 0.01 # parameter for weight decay, default, 1e-4.
+num_warmup_epochs: 0 # number of warm-up epochs before learning rate decay
+max_length: 256 # maximum length of texts
 lr: 0.00005

src/templates/template-text-classification/config.yaml

@@ -1,3 +1,3 @@
 #::= from_template_common ::#
-lr: 0.0001
-model: resnet18
+lr: 0.0001 # learning rate
+model: resnet18 # model name

src/templates/template-vision-classification/config.yaml

@@ -1,5 +1,5 @@
 #::= from_template_common ::#
-lr: 0.0001
-z_dim: 100
-d_filters: 64
-g_filters: 64
+lr: 0.0001 # learning rate
+z_dim: 100 # input dimension
+d_filters: 64 # number of filters in the first conv layer of discriminator
+g_filters: 64 # number of filters in the first conv layer of generator

src/templates/template-vision-dcgan/config.yaml

@@ -1,4 +1,4 @@
 #::= from_template_common ::#
-lr: 0.007
-accumulation_steps: 4
-num_classes: 21
+lr: 0.007 # learning rate
+accumulation_steps: 4 # parameter for gradient accumulation
+num_classes: 21 # number of classes for segmentation