From 38a8a8eecee96d9105b626ebf90acc750cf33683 Mon Sep 17 00:00:00 2001
From: Pavan Mandava <pavan@noreply.git.pavanmandava.com>
Date: Fri, 27 Jan 2023 23:46:37 +0100
Subject: [PATCH] added prompt ensemble to training script

---
 prompt-learning/train_prompting.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/prompt-learning/train_prompting.sh b/prompt-learning/train_prompting.sh
index 9d9d6af..d533221 100644
--- a/prompt-learning/train_prompting.sh
+++ b/prompt-learning/train_prompting.sh
@@ -48,7 +48,6 @@ SAVE_DIR="${SAVED_MODELS_PROMPT}"/"${experiment_folder}"
 
 echo "Trained Models (epochs) will be saved in ${SAVE_DIR}"
 
-# different number of epoch for different training sets
 # when using prompt ensemble for training, preferably use more number of epochs.
 epochs=5
 
@@ -59,5 +58,6 @@ python prompt_train.py \
 --validation_file=../data/prompt-learning/valid/valid.soloist.json \
 --num_epochs $epochs \
 --learning_rate 5e-5 \
+--with_prompt_ensemble \
 --with_inverse_prompt \
 --inverse_prompt_weight 0.3
\ No newline at end of file