Use different approach for applying max eig, with matmul, no inverse.

This commit is contained in:
Daniel Povey 2022-07-31 01:32:11 -07:00
parent ed1a147ef1
commit 90fa8a63eb

View File

@ -1821,7 +1821,7 @@ def _test_eve_cain():
fix_random_seed(42)
Linear = torch.nn.Linear if iter == 0 else ScaledLinear
hidden_dim = 200
hidden_dim = 300
m = torch.nn.Sequential(Linear(E, hidden_dim),
torch.nn.PReLU(),
Linear(hidden_dim, hidden_dim),