FIX: fixed drive mapping in train_model.sh

NINAnor · Oct 5, 2023 · 455de78 · 455de78
1 parent 4798fc4
commit 455de78
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/train_model.sh b/train_model.sh
@@ -1,9 +1,10 @@
 #!/bin/bash
 
 BASE_FOLDER=$1
-CONFIG_PATH="./CONFIG.yaml"
+CONFIG_PATH="/app/CONFIG.yaml"
 
 docker run -v $BASE_FOLDER:/data \
+            -v $PWD:/app \
             --gpus all \
-            dcase \
-            poetry run prototypicalbeats/trainer.py fit --config $CONFIG_PATH
+            beats \
+            poetry run prototypicalbeats/trainer.py fit --config $CONFIG_PATH