Fixed def file. Should run simulations now.

berenslab · Oct 17, 2024 · 0c8556e · 0c8556e · fabioseel · Oct 18, 2024
1 parent 3e03c8c
commit 0c8556e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/resources/retinal-rl.def b/resources/retinal-rl.def
@@ -47,7 +47,6 @@ From: ubuntu:22.04
   # Plotting Depends
   apt install -y \
     ffmpeg \
-    python3-pyqt5 \
     libcairo2-dev \
     fonts-liberation
 
@@ -68,6 +67,7 @@ From: ubuntu:22.04
     vizdoom==1.2.4  \
     matplotlib==3.9.1 \
     opentsne==1.0.2 \
+    opencv-python==4.10.0.84 \
     pygame==2.6.1 \
     pycairo==1.26.1  \
     git+https://github.com/pytorch/captum.git@fd758e025673100cb6a525d59a78893c558b825b \
@@ -85,4 +85,4 @@ From: ubuntu:22.04
   rm acc159linux-x64.zip
   apt clean
   rm -rf /var/lib/apt/lists/*
-  rm -rf /root/.cache/pip
+  rm -rf /root/.cache/pip
diff --git a/runner/initialize.py b/runner/initialize.py
@@ -48,7 +48,8 @@ def _initialize_create(
     # create the directories
     os.makedirs(cfg.system.data_dir)
     os.makedirs(cfg.system.checkpoint_dir)
-    os.makedirs(cfg.system.plot_dir)
+    if not cfg.use_wandb:
+        os.makedirs(cfg.system.plot_dir)
 
     # initialize the training histories
     histories: Dict[str, List[float]] = {}