From 8b5a60b03c674e080a75307c9662cde75d8f45fa Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 26 Jan 2023 14:41:22 +0900 Subject: [PATCH] from local --- .../.train_adapter.py.swp | Bin 90112 -> 90112 bytes .../train_adapter.py | 5 ++++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_adapter.py.swp index 15e53574e1ba1e09179227d179b825eb8a75dba4..1e53d130d36afd11125fa9f2693f4474b583ca0a 100644 GIT binary patch delta 229 zcmZoTz}j$tRW!*U%+puFQqO<^2m}}yVudax-`*&?TY#}}^F;yUdR`u&_!~xuWZ`DP z3ELU#{g@aSEPz-7h(7|QK++08P@Y+mp-_@vl$@bg4B|6@2oNYqttcthODrfz%}dcJ z$cYE@qf1Ll;xkiXH5tHafJ8x2W?l(F`BGJ|K8Uuo{2~R2ISQG13Q372$r;f~nJG%K LR-1DsO*snyHyc5| delta 164 zcmZoTz}j$tRW!*U%+puFQqO<^2m}}yf`u+6U)m_TTYxcd^F;yUdfrB$cqkJ@GH! diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py index 9c5bd69a6..c51e753c7 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py @@ -835,7 +835,10 @@ def compute_loss( #texts = batch["supervisions"]["text"] texts = [] - for utt_id in + for utt_id in batch["id"]: + print(utt_id) + print(pl_texts[utt_id]) + texts.append(pl_texts[utt_id]) token_ids = sp.encode(texts, out_type=int) y = k2.RaggedTensor(token_ids).to(device)