mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 10:16:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
Rezakh20 2024-03-05 12:21:30 +03:30 committed by GitHub
parent 8836e004c8
commit bf5787c438
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/WSASR/conformer_ctc2/train.py
View File

@ -382,6 +382,9 @@ def get_params() -> AttributeDict:
- warm_step: The warm_step for Noam optimizer. - warm_step: The warm_step for Noam optimizer.
""" """
parser = get_parser()
LibriSpeechAsrDataModule.add_arguments(parser)
args = parser.parse_args()
params = AttributeDict( params = AttributeDict(
{ {
"best_train_loss": float("inf"), "best_train_loss": float("inf"),
@ -394,6 +397,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,
"subsampling_factor": 2, "subsampling_factor": 2,
"encoder_dim": 512, "encoder_dim": 512,
"nhead": 8, "nhead": 8,