From 99800d13f8948f52bb587220ff6dbd102814f7ab Mon Sep 17 00:00:00 2001 From: Pavan Mandava Date: Mon, 28 Nov 2022 08:25:53 +0100 Subject: [PATCH] Updated README.md --- README.md | 2 +- prompt-learning/train_prompting.sh | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 4b51966..d1f5954 100644 --- a/README.md +++ b/README.md @@ -100,7 +100,7 @@ Edit the [set_env.sh](set_env.sh) file and set the paths (as required) for the f `OUTPUTS_DIR_PROMPT` - Path for storing the prompt model outputs (generations) -> :information_source: **Note**: Change the paths of each environment variable and make sure it matches your local system. Invalid/Wrong paths may lead to errors in the training/testing script. +> :information_source: **Note**: Change the path for each environment variable and make sure it matches with your local system. Invalid/Wrong paths may lead to errors while running the training/testing script. ```shell nano set_env.sh diff --git a/prompt-learning/train_prompting.sh b/prompt-learning/train_prompting.sh index 44d8bfe..13a8f8e 100644 --- a/prompt-learning/train_prompting.sh +++ b/prompt-learning/train_prompting.sh @@ -50,9 +50,9 @@ echo "Trained Models (epochs) will be saved in ${SAVE_DIR}" # different number of epoch for different training sets if [ "$data_split" = "5-dpd" ] || [ "$data_split" = "10-dpd" ]; then - epochs=5 + epochs=7 else - epochs=8 + epochs=10 fi python prompt_train.py \