from local

2023-01-20 13:56:53 +09:00 · 2023-01-20 13:56:53 +09:00 · 03bd69cb9f
commit 03bd69cb9f
parent 401555216a
2 changed files with 2 additions and 1 deletions
--- a/egs/librispeech/ASR/conformer_ctc2/.train.py.swp
+++ b/egs/librispeech/ASR/conformer_ctc2/.train.py.swp
--- a/egs/librispeech/ASR/conformer_ctc2/train.py
+++ b/egs/librispeech/ASR/conformer_ctc2/train.py
@ -685,7 +685,8 @@ def train_one_epoch(
        params.batch_idx_train += 1
        batch_size = len(batch["supervisions"]["text"])
        #print(batch["supervisions"])
-        batch_name = batch["supervisions"]["id"]
+        #batch_name = batch["supervisions"]["id"]
+        batch_name = str(batch_idx)

        with torch.cuda.amp.autocast(enabled=params.use_fp16):
            loss, loss_info = compute_loss(