From 9a452dd34eb9d3613a38704aa4fd7f26435fb13a Mon Sep 17 00:00:00 2001 From: ViperEkura <3081035982@qq.com> Date: Fri, 3 Oct 2025 22:08:28 +0800 Subject: [PATCH] =?UTF-8?q?fix(khaosz/trainer/data=5Futil.py):=20=E4=BF=AE?= =?UTF-8?q?=E5=A4=8D=20RandomSampler=20=E4=B8=AD=E8=BF=AD=E4=BB=A3?= =?UTF-8?q?=E8=AE=A1=E6=95=B0=E5=99=A8=E4=BD=8D=E7=BD=AE=E9=94=99=E8=AF=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- khaosz/trainer/data_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/khaosz/trainer/data_util.py b/khaosz/trainer/data_util.py index 9d6eb34..02c34cb5 100644 --- a/khaosz/trainer/data_util.py +++ b/khaosz/trainer/data_util.py @@ -297,8 +297,8 @@ class RandomSampler(Sampler[int]): start = self.current_iter % n for i in range(start, n): - yield self._indices[i] self.current_iter += 1 + yield self._indices[i] self.epoch += 1 self._indices = None