forked from huawei/mindspore2022
update graphengine & adapt to r1.0
This commit is contained in:
parent
6caf53008a
commit
cc58bda7fc
|
|
@ -1 +1 @@
|
|||
Subproject commit 63a27119bf705511946f690c55c722c356f4b397
|
||||
Subproject commit 86596e18efba333aed421fcc7aa453aa57de001a
|
||||
|
|
@ -26,9 +26,8 @@ tile_op_info = TBERegOp("Tile") \
|
|||
.attr("multiples", "optional", "listInt", "all")\
|
||||
.input(0, "x1", False, "required", "all") \
|
||||
.output(0, "y", False, "required", "all") \
|
||||
.dtype_format(DataType.I32_Default, DataType.I32_Default) \
|
||||
.dtype_format(DataType.I16_Default, DataType.I16_Default) \
|
||||
.dtype_format(DataType.F32_Default, DataType.F32_Default) \
|
||||
.op_pattern("dynamicFormat") \
|
||||
.dtype_format(DataType.None_None, DataType.None_None) \
|
||||
.get_op_info()
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -548,8 +548,6 @@ class Model:
|
|||
>>> model.train(2, dataset)
|
||||
"""
|
||||
check_bool(dataset_sink_mode)
|
||||
if sink_size == -1:
|
||||
sink_size = train_dataset.get_dataset_size()
|
||||
check_int(sink_size)
|
||||
if sink_size < -1 or sink_size == 0:
|
||||
raise ValueError("The sink_size must be -1 or positive, but got sink_size {}.".format(sink_size))
|
||||
|
|
|
|||
|
|
@ -21,7 +21,7 @@ from mindspore import Tensor
|
|||
class MindData:
|
||||
""" Stub for MindData """
|
||||
|
||||
def __init__(self, size=1, batch_size=None, repeat_count=1,
|
||||
def __init__(self, size=None, batch_size=None, repeat_count=1,
|
||||
np_types=None, output_shapes=None, input_indexs=()):
|
||||
self._size = size
|
||||
self._batch_size = batch_size
|
||||
|
|
|
|||
|
|
@ -113,7 +113,7 @@ class TrainOneStepWithLossScaleCell(nn.Cell):
|
|||
|
||||
class DatasetLenet(MindData):
|
||||
def __init__(self, predict, label, length=3):
|
||||
super(DatasetLenet, self).__init__(size=length)
|
||||
super(DatasetLenet, self).__init__()
|
||||
self.predict = predict
|
||||
self.label = label
|
||||
self.index = 0
|
||||
|
|
|
|||
Loading…
Reference in New Issue