mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typos and modify the expr of ScaledEmbedding

Browse Source
This commit is contained in:
Quandwang 2022-07-20 17:46:40 +08:00
parent 0738a44a46
commit b441dffa83
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/conformer_ctc2/train.py
View File

@ -990,7 +990,7 @@ def run(rank, world_size, args):
train_cuts, sampler_state_dict=sampler_state_dict train_cuts, sampler_state_dict=sampler_state_dict
) )
valid_cuts = librispeech.dev_cuts() valid_cuts = librispeech.dev_clean_cuts()
valid_cuts += librispeech.dev_other_cuts() valid_cuts += librispeech.dev_other_cuts()
valid_dl = librispeech.valid_dataloaders(valid_cuts) valid_dl = librispeech.valid_dataloaders(valid_cuts)

3
egs/librispeech/ASR/pruned_transducer_stateless2/scaling.py
View File

@ -643,7 +643,8 @@ class ScaledEmbedding(nn.Module):
) )
def extra_repr(self) -> str: def extra_repr(self) -> str:
s = "{num_embeddings}, {embedding_dim}, scale={scale}" # s = "{num_embeddings}, {embedding_dim}, scale={scale}"
s = "{num_embeddings}, {embedding_dim}"
if self.padding_idx is not None: if self.padding_idx is not None:
s += ", padding_idx={padding_idx}" s += ", padding_idx={padding_idx}"
if self.scale_grad_by_freq is not False: if self.scale_grad_by_freq is not False: