mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Remove unused out_balancer member
This commit is contained in:
parent
6b12f20995
commit
e5a0d8929b
@ -999,12 +999,6 @@ class Conv2dSubsampling(nn.Module):
|
|||||||
out_height = (((in_channels - 1) // 2 - 1) // 2)
|
out_height = (((in_channels - 1) // 2 - 1) // 2)
|
||||||
self.out = ScaledLinear(out_height * layer3_channels, out_channels)
|
self.out = ScaledLinear(out_height * layer3_channels, out_channels)
|
||||||
|
|
||||||
# constrain median of output to be close to zero.
|
|
||||||
self.out_balancer = ActivationBalancer(
|
|
||||||
out_channels,
|
|
||||||
channel_dim=-1, min_positive=0.45, max_positive=0.55
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def forward(self, x: torch.Tensor) -> torch.Tensor:
|
def forward(self, x: torch.Tensor) -> torch.Tensor:
|
||||||
"""Subsample x.
|
"""Subsample x.
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user