solved thor resnet50 train failed

pull/4333/head
wangmin 5 years ago
parent 13c2b23356
commit 653bcab1c7

@ -15,12 +15,19 @@
"""Dataset help for minddata dataset""" """Dataset help for minddata dataset"""
from mindspore._checkparam import check_bool from mindspore._checkparam import check_bool
from mindspore.parallel._utils import _get_device_num, _get_parallel_mode from mindspore.parallel._utils import _get_device_num, _get_parallel_mode
from mindspore.train.dataset_helper import _send_data
from mindspore.train._utils import _exec_datagraph, _get_types_and_shapes, \ from mindspore.train._utils import _exec_datagraph, _get_types_and_shapes, \
_to_full_shapes _to_full_shapes
from mindspore.train.parallel_utils import ParallelMode from mindspore.train.parallel_utils import ParallelMode
def _send_data(dataset):
"""Engine dataset to write data to tdt queue."""
if not hasattr(dataset, '__has_sent__'):
exec_dataset = dataset.__TRANSFER_DATASET__
exec_dataset.send()
dataset.__has_sent__ = True
class DatasetHelper: class DatasetHelper:
""" """
Help function to use the Minddata dataset. Help function to use the Minddata dataset.

Loading…
Cancel
Save