google-research / prompt-tuning

Original Implementation of Prompt Tuning from Lester, et al, 2021
Apache License 2.0
645 stars 58 forks source link

How to perform effective PT in LLaMA-2-13B and CNN/DM #297

Open junzhang-zj opened 7 months ago

junzhang-zj commented 7 months ago

Great job and warehouse! I set 20 virtual tokens, used lr=1e-2, and randomly selected 3k instances from the CNN/DM training set to prompt tuning LLaMA-2-13B. I've seen a steady decrease in PPL (around 2.7) on both the training and validation sets. However, its rouge score did not improve significantly on the test set. Do you have any suggestions for changing the settings of effective PT?