bug fix to sampler

Author: yhenon-nextdroid

dataloader.py

@@ -447,9 +447,9 @@ def __iter__(self):
 
  def __len__(self):
  if self.drop_last:
- return len(self.sampler) // self.batch_size
+ return len(self.data_source) // self.batch_size
  else:
- return (len(self.sampler) + self.batch_size - 1) // self.batch_size
+ return (len(self.data_source) + self.batch_size - 1) // self.batch_size
 
  def group_images(self):
  # determine the order of the images