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

check whether q k v weight is None

Browse Source
This commit is contained in:
Quandwang 2022-07-21 20:12:19 +08:00
parent a1a84ed148
commit 878026864c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/librispeech/ASR/conformer_ctc2/attention.py
View File

@ -183,13 +183,13 @@ class MultiheadAttention(nn.Module):
if not self._qkv_same_embed_dim:
q_proj_weight = (
self.q_proj_weight.get_weight() if self.q_proj_weight else None
self.q_proj_weight.get_weight() if self.q_proj_weight is not None else None
)
k_proj_weight = (
self.k_proj_weight.get_weight() if self.k_proj_weight else None
self.k_proj_weight.get_weight() if self.k_proj_weight is not None else None
)
v_proj_weight = (
self.v_proj_weight.get_weight() if self.v_proj_weight else None
self.v_proj_weight.get_weight() if self.v_proj_weight is not None else None
)
(
attn_output,