diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp index b5bc3bc15..164d78417 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.model.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py index 76eb8467c..45f46b93e 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/model.py @@ -87,6 +87,8 @@ class Transducer(nn.Module): new_emb = torch.empty(512, 50) for i in range(512): mean, std = statistic[i].strip().split(' ') + print(new_emb[i].size()) + print(float(mean), float(std)) new_emb[i] = torch.normal(float(mean), float(std), size=(50)).squeeze() new_emb = new_emb.transpose(1,0) self.prompt = torch.nn.Parameter(new_emb)