update vad_inference.py

凌匀

2023-02-16 41b1d35048603ba0a74e9dee5be1a1c8c76444d3

update vad_inference.py

 funasr/bin/vad_inference.py

@@ -111,6 +111,7 @@

        # b. Forward Encoder sreaming
        segments = []
        segments_tmp = []
        step = 6000
        t_offset = 0
        for t_offset in range(0, feats_len, min(step, feats_len - t_offset)):
@@ -128,9 +129,8 @@
            batch = to_device(batch, device=self.device)
            segments_part = self.vad_model(**batch)
            if segments_part:
                segments += segments_part
        #print(segments)

                segments_tmp += segments_part[0]
        segments.append(segments_tmp)
        return segments