From 9ffc024e3b8214d969b39d038c91cd504b5c28af Mon Sep 17 00:00:00 2001 From: fayejf <36722593+fayejf@users.noreply.github.com> Date: Mon, 12 Sep 2022 22:21:02 -0700 Subject: [PATCH] Fix and refactor label models (#4913) * fix testing after training hanging issue Signed-off-by: fayejf * fix cal label occurence Signed-off-by: fayejf * refactor loss in label_models Signed-off-by: fayejf * change langid yaml for refactor Signed-off-by: fayejf * style fix Signed-off-by: fayejf * change speaker yaml files for refactor Signed-off-by: fayejf * fix hang issue for speaker script Signed-off-by: fayejf * reflect nithin's comment and update Signed-off-by: fayejf * update lang id yaml Signed-off-by: fayejf * refactor loss instantiation Signed-off-by: fayejf * add loss to EncDecSpeakerLabelModel test Signed-off-by: fayejf * omegaconf new varible Signed-off-by: fayejf * pop Signed-off-by: fayejf * revert loss in test Signed-off-by: fayejf * fix decoder angular Signed-off-by: fayejf * fix lgtm Signed-off-by: fayejf * remove cls cfg loss check Signed-off-by: fayejf * reflect comment Signed-off-by: fayejf Signed-off-by: fayejf Signed-off-by: Matvei Novikov --- nemo/collections/asr/models/label_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo/collections/asr/models/label_models.py b/nemo/collections/asr/models/label_models.py index 5859dbfa40e0..2a5dcf1eae35 100644 --- a/nemo/collections/asr/models/label_models.py +++ b/nemo/collections/asr/models/label_models.py @@ -502,4 +502,4 @@ def get_batch_embeddings(speaker_model, manifest_filepath, batch_size=32, sample all_logits, true_labels, all_embs = np.asarray(all_logits), np.asarray(all_labels), np.asarray(all_embs) - return all_embs, all_logits, true_labels, dataset.id2label + return all_embs, all_logits, true_labels, dataset.id2label \ No newline at end of file