From ff83116369ba6afe56430cb0b9d977fbe7f4bced Mon Sep 17 00:00:00 2001
From: Atomie CHEN <atomic_cwh@163.com>
Date: 星期一, 25 三月 2024 09:37:26 +0800
Subject: [PATCH] fix: resolve IndexError when using spk model and the audio contains only 1 segment (#1535)
---
funasr/datasets/audio_datasets/samplers.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/datasets/audio_datasets/samplers.py b/funasr/datasets/audio_datasets/samplers.py
index a0ff4b6..4d78d52 100644
--- a/funasr/datasets/audio_datasets/samplers.py
+++ b/funasr/datasets/audio_datasets/samplers.py
@@ -212,7 +212,7 @@
def set_epoch(self, epoch):
self.epoch = epoch
-class CustomDistributedDynamicBatchSampler(Sampler):
+class CustomDistributedDynamicBatchSampler(DistributedSampler):
def __init__(self, dataset,
batch_size,
num_replicas=None,
--
Gitblit v1.9.1