From 64dbb9d32a51b1bce6c9de67069dc8f5943a5399 Mon Sep 17 00:00:00 2001
From: Pherkel
Date: Mon, 11 Sep 2023 22:16:26 +0200
Subject: added n_feats from config

---
 swr2_asr/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'swr2_asr/train.py')

diff --git a/swr2_asr/train.py b/swr2_asr/train.py
index eb79ee2..ca70d21 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -223,8 +223,8 @@ def main(config_path: str):
         )
     tokenizer = CharTokenizer.from_file(tokenizer_config["tokenizer_path"])
 
-    train_data_processing = DataProcessing("train", tokenizer)
-    valid_data_processing = DataProcessing("valid", tokenizer)
+    train_data_processing = DataProcessing("train", tokenizer, {"n_feats": model_config["n_feats"]})
+    valid_data_processing = DataProcessing("valid", tokenizer, {"n_feats": model_config["n_feats"]})
 
     train_loader = DataLoader(
         dataset=train_dataset,
-- 
cgit v1.2.3