forked from mindspore-Ecosystem/mindspore
Copy image data to avoid processing the same memory
This commit is contained in:
parent
be56c201ef
commit
e40ef672b7
|
@ -158,6 +158,7 @@ def _construct_input_tensors(dataset_types, dataset_shapes, device_number=1):
|
|||
def _check_to_numpy(plugin, tensor):
|
||||
"""Check the tensor and return a numpy.ndarray."""
|
||||
np_value = tensor.asnumpy()
|
||||
np_value = np_value.copy()
|
||||
if plugin == 'scalar':
|
||||
if np_value.size == 1:
|
||||
return np_value
|
||||
|
|
Loading…
Reference in New Issue