Fix repr for conv (#32897)

add nx
This commit is contained in:
Arthur 2024-08-20 14:34:24 +02:00 committed by GitHub
parent fd06ad5438
commit 65f4bc99f9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -96,6 +96,7 @@ class Conv1D(nn.Module):
def __init__(self, nf, nx):
super().__init__()
self.nf = nf
self.nx = nx
self.weight = nn.Parameter(torch.empty(nx, nf))
self.bias = nn.Parameter(torch.zeros(nf))
nn.init.normal_(self.weight, std=0.02)