From 411e7f73b271a517d23cb4a203472de97543ab66 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 14 Mar 2023 16:33:28 +0900 Subject: [PATCH] from local --- .../.train_adapter.py.swp | Bin 73728 -> 73728 bytes .../train_adapter.py | 10 +++++----- 2 files changed, 5 insertions(+), 5 deletions(-) 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 b39ecca2e41f317f51c68570c988ab0d9ffbec8e..fb96e1cc43ab7fbfde0531488cbc62f4564427bb 100644 GIT binary patch delta 145 zcmZoTz|wGlMKsAE%+puFQqO<^2m}}yl$8Zi;x>xDpUY9f^Y delta 141 zcmZoTz|wGlMKsAE%+puFQqO<^2m}}y#FPb6#5an*pUKB3P3QKXOX%Dh*4Tn5}%o(V5^W^TB4Vk!T=JR a?7K)spd_`Tq!=h39Sasy-rTsTz7_zOHzC9T 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 ef6901250..5ec2d89ba 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 @@ -833,11 +833,11 @@ def compute_loss( batch_idx_train = params.batch_idx_train warm_step = params.warm_step - #texts = batch["supervisions"]["text"] - texts = [] - for cut in supervisions['cut']: - utt_id = cut.id - texts.append(pl_texts[utt_id]) + texts = batch["supervisions"]["text"] + #texts = [] + #for cut in supervisions['cut']: + # utt_id = cut.id + # texts.append(pl_texts[utt_id]) token_ids = sp.encode(texts, out_type=int) y = k2.RaggedTensor(token_ids).to(device)