Browse Source

!3304 fix dataset iterator bug

Merge pull request !3304 from jinyaohui/master
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
78d0ced83e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/dataset_helper.py

+ 1
- 1
mindspore/train/dataset_helper.py View File

@@ -222,9 +222,9 @@ class _DatasetIterNormal:
self.dataset = dataset
self.device_num = _get_device_num()
self.global_rank = _get_global_rank()
self.iter = self.dataset.create_tuple_iterator()

def __iter__(self):
self.iter = self.dataset.create_tuple_iterator()
return self

def __next__(self):


Loading…
Cancel
Save