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

style correction

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

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

@ -183,13 +183,19 @@ 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 is not None 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 is not None 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 is not None else None
self.v_proj_weight.get_weight()
if self.v_proj_weight is not None
else None
)
(
attn_output,