From f2ed4b3856eaed8abe568e6904ffd8dc3a799f5f Mon Sep 17 00:00:00 2001
From: 彭震东 <zhendong.peng@qq.com>
Date: 星期一, 15 七月 2024 17:54:27 +0800
Subject: [PATCH] fix progress bar for batch_size (#1917)
---
funasr/models/transformer/model.py | 2 --
1 files changed, 0 insertions(+), 2 deletions(-)
diff --git a/funasr/models/transformer/model.py b/funasr/models/transformer/model.py
index 0d5ed23..adfd525 100644
--- a/funasr/models/transformer/model.py
+++ b/funasr/models/transformer/model.py
@@ -145,8 +145,6 @@
text: (Batch, Length)
text_lengths: (Batch,)
"""
- # import pdb;
- # pdb.set_trace()
if len(text_lengths.size()) > 1:
text_lengths = text_lengths[:, 0]
if len(speech_lengths.size()) > 1:
--
Gitblit v1.9.1