fix(data): 修改 Sampler 的长度计算方式, 避免提前初始化

This commit is contained in:
ViperEkura 2025-12-10 18:57:53 +08:00
parent 8e975017d3
commit a30ddca517
1 changed files with 1 additions and 3 deletions

View File

@ -75,6 +75,4 @@ class ResumableDistributedSampler(Sampler[int]):
self._indices = None
def __len__(self):
if self._indices is None:
self._get_indices()
return len(self._indices)
return self.num_samples_per_replica