pminervini commited on
Commit
e598f52
1 Parent(s): b8ff31e
Files changed (1) hide show
  1. src/backend/envs.py +3 -2
src/backend/envs.py CHANGED
@@ -32,9 +32,10 @@ class Tasks(Enum):
32
  # task6 = Task("halueval_dialogue", "acc", "HaluEval Dialogue", 0)
33
  # task7 = Task("halueval_summarization", "acc", "HaluEval Summarization", 0)
34
 
35
- # task6 = Task("xsum", "rougeL_acc", "XSum", 8)
 
36
 
37
- task8 = Task("memo-trap", "acc", "memo-trap", 0)
38
 
39
  # NUM_FEWSHOT = 64 # Change with your few shot
40
 
 
32
  # task6 = Task("halueval_dialogue", "acc", "HaluEval Dialogue", 0)
33
  # task7 = Task("halueval_summarization", "acc", "HaluEval Summarization", 0)
34
 
35
+ task8 = Task("xsum", "rougeL", "XSum", 2)
36
+ task9 = Task("cnndm", "rougeL", "CNN/DM", 2)
37
 
38
+ task10 = Task("memo-trap", "acc", "memo-trap", 0)
39
 
40
  # NUM_FEWSHOT = 64 # Change with your few shot
41