From 4daea3711063c64485be3c00eaa9727404549f51 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 24 二月 2023 17:55:00 +0800
Subject: [PATCH] onnx
---
funasr/export/models/predictor/cif.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/funasr/export/models/predictor/cif.py b/funasr/export/models/predictor/cif.py
index 5518cb8..fcfcd5f 100644
--- a/funasr/export/models/predictor/cif.py
+++ b/funasr/export/models/predictor/cif.py
@@ -109,7 +109,8 @@
frames = torch.stack(list_frames, 1)
list_ls = []
len_labels = torch.round(alphas.sum(-1)).int()
- max_label_len = len_labels.max()
+ max_label_len = len_labels.max().item()
+ print("type: {}".format(type(max_label_len)))
for b in range(batch_size):
fire = fires[b, :]
l = torch.index_select(frames[b, :, :], 0, torch.nonzero(fire >= threshold).squeeze())
--
Gitblit v1.9.1