from local
This commit is contained in:
parent
a433f440bb
commit
6ffc0bcd5b
Binary file not shown.
@ -43,6 +43,8 @@ class ConvolutionModule(nn.Module):
|
|||||||
bias=bias,
|
bias=bias,
|
||||||
)
|
)
|
||||||
self.norm = nn.BatchNorm1d(channels)
|
self.norm = nn.BatchNorm1d(channels)
|
||||||
|
self.layer_norm = nn.LayerNorm(channels)
|
||||||
|
|
||||||
self.pointwise_conv2 = nn.Conv1d(
|
self.pointwise_conv2 = nn.Conv1d(
|
||||||
channels,
|
channels,
|
||||||
channels,
|
channels,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user