update bash script

ferugit · Mar 1, 2023 · dd7b7a2 · dd7b7a2
1 parent adb07b5
commit dd7b7a2
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/train_model.sh b/train_model.sh
@@ -11,9 +11,17 @@ epochs=80
 batch_size=12
 patience=10
 lr=0.001
-
+cuda=True
+balance=True
 # arch
 # augments
-# etc
+
+# Audio config
+window_size=3.0
+sampling_rate=16000
+
+# Dataset
+num_classes=5
+
 
 python -u src/train_and_evaluate.py