Browse Source

!1609 fix subset random sampler error

Merge pull request !1609 from yanghaitao/yht_subsetrandomsampler
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
83b74e4693
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