From 9afe6f5e0d23a35e9d60e9e7224c7f5858796d53 Mon Sep 17 00:00:00 2001 From: Jocelyn Huang Date: Mon, 25 Sep 2023 14:27:34 -0700 Subject: [PATCH] Fix get_dist() tensor dimension Signed-off-by: Jocelyn Huang --- nemo/collections/tts/modules/aligner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo/collections/tts/modules/aligner.py b/nemo/collections/tts/modules/aligner.py index 2910602474fd..f044a86a52eb 100644 --- a/nemo/collections/tts/modules/aligner.py +++ b/nemo/collections/tts/modules/aligner.py @@ -98,7 +98,7 @@ def get_dist(self, keys, queries, mask=None): self._apply_mask(dist, mask, float("inf")) - return dist + return dist.squeeze(1) @staticmethod def get_euclidean_dist(queries_enc, keys_enc):