Updated requirements and fixed some scripts

.gitignore

@@ -179,3 +179,6 @@ ipython_config.py
 # Remove previous ipynb_checkpoints
 #   git rm -r .ipynb_checkpoints/
 
+# runs folder
+baseline/runs/
+

README.md

@@ -148,4 +148,4 @@ python evaluate.py
 | 125-dpd | 35.79 |
 | 250-dpd | 40.38 |
 
-> Note: The above results might change based on further experiments  
+> Note: The above preliminary results will change based on further experiments  

baseline/requirements.txt

@@ -1,6 +1,6 @@
 numpy==1.17.4
 torch==1.6.0
 tqdm==4.38.0
-transformers==2.1.1
+transformers==2.2.2
 tensorboard==2.1.1
 tensorboardx==2.0

baseline/soloist_decode.py

@@ -9,10 +9,6 @@ import torch
 import torch.nn.functional as F
 import numpy as np
 
-import sys
-sys.path.append('.')
-sys.path.append('./transformers')
-sys.path.append('./transformers/')
 
 from transformers import GPT2Config
 from transformers import GPT2LMHeadModel, GPT2Tokenizer

baseline/soloist_train.py

@@ -11,11 +11,6 @@ import shutil
 import time
 import json
 
-import sys
-# sys.path.append('.')
-# sys.path.append('./transformers')
-# sys.path.append('./transformers/')
-
 import numpy as np
 import torch
 from torch.utils.data import DataLoader, Dataset, SequentialSampler, RandomSampler

baseline/train_baseline.sh

@@ -34,7 +34,7 @@ SAVE_DIR="${SAVED_MODELS_BASELINE}"/"${experiment_folder}"
 mkdir -p "${SAVE_DIR}"
 
 echo "Trained Model checkpoints are saved in ${SAVE_DIR}"
-echo "Cleaning the directory (${experiment_folder}) to make sure it's empty!"
+echo "Cleaning the experiment directory (${experiment_folder}) to make sure it's empty!"
 
 # delete the contents of the dir (where model checkpoints will be saved!)
 rm -rf "${SAVE_DIR}"