Added test in multiclass.yaml, multilabel.yaml

Author: nikiduki

tests/assets/configs/multiclass.yaml

@@ -35,6 +35,12 @@
       batch_size: [8, 16]
       learning_rate: [5.0e-5]
       seed: [0]
+    - module_name: ptuning
+      base_model_config: ["prajjwal1/bert-tiny"]
+      num_train_epochs: [1]
+      batch_size: [8, 16]
+      task_type: ["SEQ_CLS"]
+      num_virtual_tokens: [10, 20]
 - node_type: decision
   target_metric: decision_accuracy
   search_space:

tests/assets/configs/multilabel.yaml

@@ -31,6 +31,12 @@
       batch_size: [8]
       learning_rate: [5.0e-5]
       seed: [0]
+    - module_name: ptuning
+      base_model_config: ["prajjwal1/bert-tiny"]
+      num_train_epochs: [1]
+      batch_size: [8]
+      task_type: ["SEQ_CLS"]
+      num_virtual_tokens: [10, 20]
 - node_type: decision
   target_metric: decision_accuracy
   search_space: