openprompt

1
dataset:
2
  name: agnews
3
  path: datasets/TextClassification/agnews
4

5
plm:
6
  model_name: bert
7
  model_path: bert-base-cased
8
  optimize:
9
    freeze_para: False
10
    lr: 0.00003
11
    weight_decay: 0.01
12
    scheduler:
13
      type: 
14
      num_warmup_steps: 500
15

16
train:
17
  batch_size: 2
18

19
test:
20
  batch_size: 8
21

22
dev:
23
  batch_size: 8
24

25

26

27
template: ptuning_template
28
verbalizer: manual_verbalizer
29

30

31
ptuning_template:
32
  choice: 1
33
  file_path: scripts/TextClassification/agnews/ptuning_template.txt
34

35
manual_verbalizer:
36
  choice: 0
37
  file_path: scripts/TextClassification/agnews/multiwords_verbalizer.jsonl
38
  
39
environment:
40
  num_gpus: 1
41
  cuda_visible_devices:
42
  local_rank: 0 
43

44
learning_setting: few_shot
45

46
few_shot:
47
  parent_config: learning_setting
48
  few_shot_sampling: sampling_from_train
49
  
50
sampling_from_train:
51
  parent_config: few_shot_sampling
52
  num_examples_per_label: 10
53
  also_sample_dev: True
54
  num_examples_per_label_dev: 10
55
  seed:
56
    - 123
57
    - 456
58
    - 789
59
    - 321
60
    - 654
61

62