diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py index 5b960e30e..1d87f3f73 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/optim.py @@ -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)