added prompt ensemble to training script

main
Pavan Mandava 3 years ago
parent aacf5fd181
commit 38a8a8eece

@ -48,7 +48,6 @@ SAVE_DIR="${SAVED_MODELS_PROMPT}"/"${experiment_folder}"
echo "Trained Models (epochs) will be saved in ${SAVE_DIR}" 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. # when using prompt ensemble for training, preferably use more number of epochs.
epochs=5 epochs=5
@ -59,5 +58,6 @@ python prompt_train.py \
--validation_file=../data/prompt-learning/valid/valid.soloist.json \ --validation_file=../data/prompt-learning/valid/valid.soloist.json \
--num_epochs $epochs \ --num_epochs $epochs \
--learning_rate 5e-5 \ --learning_rate 5e-5 \
--with_prompt_ensemble \
--with_inverse_prompt \ --with_inverse_prompt \
--inverse_prompt_weight 0.3 --inverse_prompt_weight 0.3
Loading…
Cancel
Save