Browse Source

!1610 fix subset random sampler error

Merge pull request !1610 from yanghaitao/yht_subsetrandomsampler_r0.3
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
9274daec9c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/dataset/engine/samplers.py

+ 1
- 1
mindspore/dataset/engine/samplers.py View File

@@ -483,7 +483,7 @@ class SubsetRandomSampler(BuiltinSampler):


def get_dataset_size(self):
return len(indices)
return len(self.indices)


class WeightedRandomSampler(BuiltinSampler):


Loading…
Cancel
Save