Update config for text-classification (#325)

* Update config for text-classification

Update config for text-classification for issue #293

* Update config.yaml

* Update config.yaml

* Update config.yaml

Author: puhuk

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

@@ -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