diff --git a/egs/librispeech/ASR/.run_v2.sh.swp b/egs/librispeech/ASR/.run_v2.sh.swp new file mode 100644 index 000000000..3f9f6d358 Binary files /dev/null and b/egs/librispeech/ASR/.run_v2.sh.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v/.train.py.swp index b98548e1f..dd903968c 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v/.train.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v/.train.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp new file mode 100644 index 000000000..9d40a7e64 Binary files /dev/null and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py index 162ad8412..049705b91 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -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