From dd7b7a23a5917599db9456093ca0060b9f4243ff Mon Sep 17 00:00:00 2001
From: ferugit <wiliam.lopezgavilanez@telefonica.com>
Date: Wed, 1 Mar 2023 16:03:07 +0100
Subject: [PATCH] update bash script

---
 train_model.sh | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/train_model.sh b/train_model.sh
index fb4eeb8..1426d21 100644
--- 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
\ No newline at end of file