mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Use different approach for applying max eig, with matmul, no inverse.
This commit is contained in:
parent
ed1a147ef1
commit
90fa8a63eb
@ -1821,7 +1821,7 @@ def _test_eve_cain():
|
|||||||
fix_random_seed(42)
|
fix_random_seed(42)
|
||||||
Linear = torch.nn.Linear if iter == 0 else ScaledLinear
|
Linear = torch.nn.Linear if iter == 0 else ScaledLinear
|
||||||
|
|
||||||
hidden_dim = 200
|
hidden_dim = 300
|
||||||
m = torch.nn.Sequential(Linear(E, hidden_dim),
|
m = torch.nn.Sequential(Linear(E, hidden_dim),
|
||||||
torch.nn.PReLU(),
|
torch.nn.PReLU(),
|
||||||
Linear(hidden_dim, hidden_dim),
|
Linear(hidden_dim, hidden_dim),
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user