fix(data): 修改 Sampler 的长度计算方式, 避免提前初始化
This commit is contained in:
parent
8e975017d3
commit
a30ddca517
|
|
@ -75,6 +75,4 @@ class ResumableDistributedSampler(Sampler[int]):
|
||||||
self._indices = None
|
self._indices = None
|
||||||
|
|
||||||
def __len__(self):
|
def __len__(self):
|
||||||
if self._indices is None:
|
return self.num_samples_per_replica
|
||||||
self._get_indices()
|
|
||||||
return len(self._indices)
|
|
||||||
Loading…
Reference in New Issue