From b7f36b40a1a7a7b13277e7c7aa39860803641ac0 Mon Sep 17 00:00:00 2001
From: Rezakh20 <160485045+Rezakh20@users.noreply.github.com>
Date: Tue, 5 Mar 2024 12:23:10 +0330
Subject: [PATCH] Update train.py

---
 egs/librispeech/WSASR/conformer_ctc2/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librispeech/WSASR/conformer_ctc2/train.py b/egs/librispeech/WSASR/conformer_ctc2/train.py
index 42e14b587..1552391fd 100755
--- a/egs/librispeech/WSASR/conformer_ctc2/train.py
+++ b/egs/librispeech/WSASR/conformer_ctc2/train.py
@@ -397,7 +397,7 @@ def get_params() -> AttributeDict:
             "valid_interval": 800,  # For the 100h subset, use 800
             "alignment_interval": 25,
             # parameters for conformer
-            "feature_dim": 768,
+            "feature_dim": args.feature_dim,
             "subsampling_factor": 2,
             "encoder_dim": 512,
             "nhead": 8,