Update train.py

change feature_dim value
This commit is contained in:
Rezakh20 2024-03-04 12:36:28 +03:30 committed by GitHub
parent 29b195a42e
commit 267a36e6ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -385,7 +385,7 @@ def get_params() -> AttributeDict:
"valid_interval": 800, # For the 100h subset, use 800 "valid_interval": 800, # For the 100h subset, use 800
"alignment_interval": 25, "alignment_interval": 25,
# parameters for conformer # parameters for conformer
"feature_dim": 768, "feature_dim": 80, # when using fbank features for training
"subsampling_factor": 2, "subsampling_factor": 2,
"encoder_dim": 512, "encoder_dim": 512,
"nhead": 8, "nhead": 8,