22 Commits

Author SHA1 Message Date
Daniel Povey
ebc2ffeff7 Bug fix 2022-05-18 10:26:29 +08:00
Daniel Povey
38ccbb6062 Think it works now.. will move to different branch though. 2022-05-18 10:11:20 +08:00
Daniel Povey
3d35018ce0 Just adding some draft code, will move this around. 2022-05-17 21:00:36 +08:00
Daniel Povey
a46f74feb4 Broadcast transformation 2022-05-17 17:23:28 +08:00
Daniel Povey
ceb4eb4b85 Scale covariance 2022-05-17 15:32:06 +08:00
Daniel Povey
c923b5900e Symmetrize covariance 2022-05-17 15:14:23 +08:00
Daniel Povey
9133b57808 Use torch.no_grad() for stats 2022-05-17 14:49:00 +08:00
Daniel Povey
07d3369234 Finish feat-diagonalizing code 2022-05-17 14:13:56 +08:00
Daniel Povey
2f2934a115 fix train.py 2022-05-16 19:41:16 +08:00
Daniel Povey
d61c8aa3bc Hopefully this finishes the full orthogonalization. 2022-05-16 19:18:12 +08:00
Daniel Povey
67f916e599 Draft towards 2nd orthogonalization 2022-05-16 16:16:12 +08:00
Daniel Povey
8aeaf1421a Rename orthogonalize to diagonalize 2022-05-16 12:43:47 +08:00
Daniel Povey
9859e33c06 Remove optimizer.reset() which is not supported for Eve 2022-05-16 12:13:36 +08:00
Daniel Povey
d58debdb3f Use Eve with orthogonalization 2022-05-16 10:34:25 +08:00
Daniel Povey
995371ad95 Move train.py changes to the right dir 2022-05-15 22:17:08 +08:00
Daniel Povey
bb32556f9e Add and test reset() function 2022-05-15 16:20:10 +08:00
Daniel Povey
a1dc020270 train scalars slower 2022-05-15 15:44:56 +08:00
Daniel Povey
5e3bf4ce5a Remove some debug code 2022-05-15 14:22:12 +08:00
Daniel Povey
6306f24430 Some changes to algorithm; more diagnostics printing 2022-05-15 14:12:23 +08:00
Daniel Povey
67c402a369 Add some debugging/diagnostic code 2022-05-15 13:28:00 +08:00
Daniel Povey
747960677e Prob. 1st working version of Abel 2022-05-15 10:13:06 +08:00
Daniel Povey
a1c6bae5d6 Init pruned_transducer_stateless4b as copy of 4 2022-05-13 22:48:17 +08:00