fix #1

scripts/eval.py

@@ -36,7 +36,7 @@ def main(hparams):
     logger.info(torch.cuda.get_device_properties(device))
     logger.info(f'Hyperparameters: \n {hparams}')
 
-    hparams.DATASET.NUM_WORKERS = -1  # set this to be compatible with other machines
+    hparams.DATASET.NUM_WORKERS = 0  # set this to be compatible with other machines
     model = PARETrainer(hparams=hparams).to(device)
 
     if hparams.TRAINING.PRETRAINED_LIT is not None:

scripts/install_conda.sh

@@ -12,4 +12,5 @@ conda activate $CONDA_ENV_NAME
 which python
 which pip
 
+sudo apt-get install libturbojpeg
 pip install -r requirements.txt

scripts/install_pip.sh

@@ -7,4 +7,5 @@ echo "Activating virtual environment"
 
 source $PWD/pare-env/bin/activate
 
+sudo apt-get install libturbojpeg
 $PWD/pare-env/bin/pip install -r requirements.txt