Add dependencies

README.md

@@ -18,9 +18,12 @@ data folder contains multiple folders and files:
     - use current path as `pretrained_model` variable in script files
 2. Clone the repository
     - git clone https://github.com/EhsanMashhadi/MSR2021-ProgramRepair.git
-3. Train the model with MSR data
+3. Install dependencies
+    - pip install torch==1.4.0
+    - pip install transformers==2.5.0
+4. Train the model with MSR data
     - bash ./scripts/codebert/train.sh
-4. Evaluate the model
+5. Evaluate the model
     - bash ./scripts/codebert/test.sh 
 
 ### Running Simple LSTM Experiments

scripts/codebert/test.sh

@@ -13,8 +13,9 @@ test_file=$data_dir/src-test.txt,$data_dir/tgt-test.txt
 test_model=$output_dir/checkpoint-best-ppl/pytorch_model.bin
 pretrained_model=./code-bert #CodeBert model path downloaded from Huggingface
 pretrained_model=../../codebert-model/codebert-base
+CodeBERT=../../codebert
 
-python ../codebert/run.py \
+python $CodeBERT/run.py \
 --do_test \
 --model_type roberta \
 --model_name_or_path $pretrained_model \

scripts/codebert/train.sh

@@ -15,8 +15,10 @@ train_file=$data_dir/src-train.txt,$data_dir/tgt-train.txt
 validate_file=$data_dir/src-val.txt,$data_dir/tgt-val.txt
 pretrained_model=./code-bert #CodeBert model path downloaded from Huggingface
 pretrained_model=../../codebert-model/codebert-base
+CodeBERT=../../codebert
 
-python ../codebert/run.py \
+
+python $CodeBERT/run.py \
 --do_train \
 --do_eval \
 --model_type roberta \