PTG.md
December 20, 2022 ยท View on GitHub
PTG
First, you may copy the prompt_source.pth into the TextBox folder (i.e.,
you can conduct the cross-dataset experiments on cnndm dataset using this command:
python run_textbox.py --model=PTG --dataset=cnndm --model_path=facebook/bart-large
In this default case, the source tasks (datasets) is msn, mn, and nr.
You can use --dataset=xxx to specify the dataset name.
In addition, you can also specify the source tasks using --source_task=list_of_task. The default setting is equivalent to --source_task=\[\'msn\',\'mn\',\'nr\'\].
There are also several cases used in the paper:
--source_task=cross-dataset2: tc, da, mw.--source_task=cross-task1: squad, wiki, quora, wp, cnndm.--source_task=cross-task2: squad, wiki, quora, wp, pc.