Cosmetic fix

This commit is contained in:
Daniel Povey 2022-11-14 23:17:20 +08:00
parent ff6431ed0f
commit 46bd93b792

View File

@ -1193,7 +1193,7 @@ class SelfAttention(nn.Module):
Args:
embed_dim: the input and output embedding dimension
num_heads: the number of attention heads
value_dim: the value dimension per head
value_head_dim: the value dimension per head
"""
def __init__(
self,