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-09 17:07:27 +09:00
parent a1a1964b95
commit 217bd54639
4 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/.run_v2.sh.swp Normal file
View File

Binary file not shown.

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

Binary file not shown.

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

Binary file not shown.

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

@ -459,7 +459,8 @@ def get_encoder_model(params: AttributeDict) -> nn.Module:
# TODO: We can add an option to switch between Zipformer and Transformer
def to_int_tuple(s: str):
return tuple(map(int, s.split(",")))
'''
encoder = Zipformer(
num_features=params.feature_dim,
output_downsampling_factor=2,
@ -474,6 +475,15 @@ def get_encoder_model(params: AttributeDict) -> nn.Module:
cnn_module_kernels=to_int_tuple(params.cnn_module_kernels),
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,
)
return encoder