mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-19 05:54:20 +00:00
Remove ActivationBalancer, unnecessary
This commit is contained in:
parent
7993c84cd6
commit
3468c3aa5a
@ -1664,10 +1664,8 @@ def _test_eve_cain():
|
||||
Linear = torch.nn.Linear if iter == 0 else ScaledLinear
|
||||
# TODO: find out why this is not converging...
|
||||
m = torch.nn.Sequential(Linear(E, 200),
|
||||
ActivationBalancer(-1),
|
||||
torch.nn.PReLU(),
|
||||
Linear(200, 200),
|
||||
ActivationBalancer(-1),
|
||||
torch.nn.PReLU(),
|
||||
Linear(200, E),
|
||||
).to(device)
|
||||
|
Loading…
x
Reference in New Issue
Block a user