fix(data): 修改 Sampler 的长度计算方式, 避免提前初始化
This commit is contained in:
parent
8e975017d3
commit
a30ddca517
|
|
@ -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
|
||||
Loading…
Reference in New Issue