mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-01-09 20:03:52 +09:00
parent 62bcb44876
commit c3201adb5a
2 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/incremental_transf/.identity_train.py.swp
View File

Binary file not shown.

14
egs/librispeech/ASR/incremental_transf/identity_train.py
View File

@ -499,16 +499,12 @@ def get_transducer_model(params: AttributeDict) -> nn.Module:
def get_interformer_model(pt_encoder, params: AttributeDict) -> nn.Module: def get_interformer_model(pt_encoder, params: AttributeDict) -> nn.Module:
encoder = get_encoder_model(params) encoder = get_encoder_model(params)
model = Transducer( model = Interformer(
encoder=encoder, pt_encoder = pt_encoder,
decoder=decoder, inter_encoder = encoder,
joiner=joiner,
encoder_dim=params.encoder_dim,
decoder_dim=params.decoder_dim,
joiner_dim=params.joiner_dim,
vocab_size=params.vocab_size,
) )
return model return model