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