Update train.py

This commit is contained in:
Rezakh20 2024-03-05 12:23:10 +03:30 committed by GitHub
parent bf5787c438
commit b7f36b40a1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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,