From c923b5900e2dcdcaf17c0a90addc0be04f9f6b23 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 17 May 2022 15:14:23 +0800 Subject: [PATCH] Symmetrize covariance --- .../ASR/pruned_transducer_stateless4b/diagonalize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless4b/diagonalize.py b/egs/librispeech/ASR/pruned_transducer_stateless4b/diagonalize.py index ec2bc9ceb..0aa0ef05a 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless4b/diagonalize.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless4b/diagonalize.py @@ -259,5 +259,5 @@ class OrthogonalTransformation(nn.Module): @torch.no_grad() def get_transformation_out(self) -> Tensor: # see also get_transformation() above for notes on this. - cov = self.feats_cov + cov = 0.5 * (self.feats_cov + self.feats_cov.t()) # make sure symmetric return get_transformation(cov)