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 2022-12-10 14:12:26 +09:00
parent 7a100fa97d
commit bc4a865377
3 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/.run_v3.sh.swp
View File

Binary file not shown.

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp
View File

Binary file not shown.

15
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py
View File

@ -550,13 +550,14 @@ def get_encoder_model(params: AttributeDict) -> nn.Module:
num_encoder_layers=to_int_tuple(params.num_encoder_layers),
)
'''
encoder = FairSeqData2VecEncoder(
input_size=768, #params.encoder_dim,
w2v_url='None',
output_size=768, #params.encoder_dim,
freeze_finetune_updates=0, #params.freeze_finetune_updates*params.accum_grads,
additional_block=True, #params.additional_block,
)
if params.encoder_type == 'd2v':
encoder = FairSeqData2VecEncoder(
input_size=params.encoder_dim,
w2v_url='None',
output_size=params.encoder_dim,
freeze_finetune_updates=params.freeze_finetune_updates*params.accum_grads,
additional_block=True, #params.additional_block,
)
return encoder