|
|
@@ -115,6 +115,8 @@ class Sampler: |
|
|
return self.child_sampler.is_sharded() |
|
|
return self.child_sampler.is_sharded() |
|
|
|
|
|
|
|
|
def get_num_samples(self): |
|
|
def get_num_samples(self): |
|
|
|
|
|
if self.num_samples is None: |
|
|
|
|
|
return None |
|
|
return self._get_indices().size |
|
|
return self._get_indices().size |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|