Updated README.md

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

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 \