[ADD] helpers bash scripts to run the models

shell_scripts/train_baseline.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+cd ..
+
+BASE_FOLDER=$1
+CONFIG_PATH="/app/CONFIG.yaml"
+
+# Check if BASE_FOLDER is not set or empty
+if [ -z "$BASE_FOLDER" ]; then
+    echo "Error: BASE_FOLDER is not specified."
+    exit 1
+fi
+
+docker run -v $BASE_FOLDER:/data -v $PWD:/app --gpus all beats poetry run prototypicalbeats/trainer.py fit \
+    --config $CONFIG_PATH \
+    --model.model_type baseline \
+    --model.state train \
+    --trainer.default_root_dir "/data/lightning_logs/baseline"

shell_scripts/train_beats.sh

@@ -0,0 +1,12 @@
+#!/bin/bash
+
+BASE_FOLDER=$1
+CONFIG_PATH="/app/CONFIG.yaml"
+
+# Check if BASE_FOLDER is not set or empty
+if [ -z "$BASE_FOLDER" ]; then
+    echo "Error: BASE_FOLDER is not specified."
+    exit 1
+fi
+
+docker run -v $BASE_FOLDER:/data -v $PWD:/app --gpus all beats poetry run prototypicalbeats/trainer.py fit --config $CONFIG_PATH 

shell_scripts/train_pann.sh

@@ -0,0 +1,18 @@
+#!/bin/bash
+cd ..
+
+BASE_FOLDER=$1
+CONFIG_PATH="/app/CONFIG.yaml"
+
+# Check if BASE_FOLDER is not set or empty
+if [ -z "$BASE_FOLDER" ]; then
+    echo "Error: BASE_FOLDER is not specified."
+    exit 1
+fi
+
+docker run -v $BASE_FOLDER:/data -v $PWD:/app --gpus all beats poetry run prototypicalbeats/trainer.py fit \
+    --config $CONFIG_PATH \
+    --model.model_type pann \
+    --model.state train \
+    --model.model_path "/data/models/PANN/Cnn14_mAP=0.431.pth" \
+    --trainer.default_root_dir "/data/lightning_logs/pann"

shell_scripts/validate_beats.sh

@@ -0,0 +1,15 @@
+#!/bin/bash
+
+BASE_DIR=$1
+
+cd ..
+
+docker run -v $BASE_DIR:/data -v $PWD:/app \
+            --gpus all \
+            beats \
+            poetry run python /app/evaluate/evaluateDCASE.py \
+            --config "/app/CONFIG_PREDICT.yaml" \
+            --overwrite 
+            #data.status="validate" \
+            #model.model_type="beats" \
+            #model.model_path="/data/lightning_logs/BEATs/lightning_logs/version_2/checkpoints/epoch=99-step=10000.ckpt"