From d1efd59af963a25314dbbe254d298ed441695ca1 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期四, 25 一月 2024 14:32:54 +0800
Subject: [PATCH] update docker lists
---
funasr/datasets/audio_datasets/samplers.py | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/funasr/datasets/audio_datasets/samplers.py b/funasr/datasets/audio_datasets/samplers.py
index 4af35e9..0d93098 100644
--- a/funasr/datasets/audio_datasets/samplers.py
+++ b/funasr/datasets/audio_datasets/samplers.py
@@ -13,6 +13,7 @@
buffer_size: int = 30,
drop_last: bool = False,
shuffle: bool = True,
+ is_training: bool = True,
**kwargs):
self.drop_last = drop_last
@@ -24,10 +25,10 @@
self.buffer_size = buffer_size
self.max_token_length = kwargs.get("max_token_length", 5000)
self.shuffle_idx = np.arange(self.total_samples)
- self.shuffle = shuffle
+ self.shuffle = shuffle and is_training
def __len__(self):
- return self.total_samples
+ return (self.total_samples-1) // self.batch_size + 1
def set_epoch(self, epoch):
np.random.seed(epoch)
--
Gitblit v1.9.1