From 03f084dc3bd930a10570bf576463b48e2eada8e8 Mon Sep 17 00:00:00 2001
From: Muhammed Kocabas <muhammed.kocabas@tuebingen.mpg.de>
Date: Thu, 14 Oct 2021 09:53:24 +0200
Subject: [PATCH] fix https://github.com/mkocabas/PARE/issues/1

---
 scripts/eval.py          | 2 +-
 scripts/install_conda.sh | 1 +
 scripts/install_pip.sh   | 1 +
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/eval.py b/scripts/eval.py
index 9821964..54f3e08 100644
--- a/scripts/eval.py
+++ b/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:
diff --git a/scripts/install_conda.sh b/scripts/install_conda.sh
index ab30bd4..2eeafc2 100644
--- a/scripts/install_conda.sh
+++ b/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
\ No newline at end of file
diff --git a/scripts/install_pip.sh b/scripts/install_pip.sh
index daa6817..e3677a1 100644
--- a/scripts/install_pip.sh
+++ b/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
\ No newline at end of file