Change ratio in NonlinAttentionModule from 8 to 2

This commit is contained in:
Daniel Povey 2022-11-29 15:38:13 +08:00
parent 36a2f33a6f
commit 753269668a

View File

@ -1459,7 +1459,7 @@ class NonlinAttentionModule(nn.Module):
""" """
def __init__( def __init__(
self, channels: int, ratio: int = 8, self, channels: int, ratio: int = 2,
) -> None: ) -> None:
super().__init__() super().__init__()