openprompt

1
dataset:
2
  name: webnlg
3
  path: ./datasets/CondGen/webnlg_2017
4

5
task: generation
6

7
train:
8
  num_epochs: 5
9
  batch_size: 2
10
  teacher_forcing: True
11
  gradient_accumulation_steps: 2
12

13
generation: # Adding any arguments for generation here.
14
  parent_config: task
15
  max_length: 512
16

17
plm:
18
  model_name: gpt2
19
  model_path: gpt2-medium
20
  optimize: 
21
    freeze_para: False
22

23
## LEARINING SETTING  ####################################################
24
learning_setting: full # selecting from "full", "zero_shot", "few_shot"
25

26
few_shot:
27
  parent_config: learning_setting
28
  few_shot_sampling: sampling_from_train
29
  
30
sampling_from_train:
31
  parent_config: few_shot_sampling
32
  num_examples_per_label: 100
33
  also_sample_dev: True
34
  num_examples_per_label_dev: 100
35
  seed:
36
    - 123
37

38
template: manual_template
39
verbalizer: 
40

41
manual_template:
42
  choice: 0
43
  file_path: scripts/CondGen/webnlg_2017/manual_template.txt
44

45

46

47

48

49

50