mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

delete unused param in device_que

This commit is contained in:
Xiao Tianci 2021-06-28 09:56:40 +08:00
parent 75f772b7df
commit 27406daf3a
2 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/dataset/engine/datasets.py
View File

@ -1258,13 +1258,11 @@ class Dataset:
return dataset
@check_device_send
def device_que(self, prefetch_size=None, send_epoch_end=True, create_data_info_queue=False):
def device_que(self, send_epoch_end=True, create_data_info_queue=False):
"""
Return a transferred Dataset that transfers data through a device.
Args:
prefetch_size (int, optional): Prefetch number of records ahead of the
user's request (default=None).
send_epoch_end (bool, optional): Whether to send end of sequence to device or not (default=True).
create_data_info_queue (bool, optional): Whether to create queue which stores
types and shapes of data or not(default=False).
@ -2444,7 +2442,6 @@ def _pyfunc_worker_exec(index, qid, *args):
return result
# PythonCallable wrapper for multiprocess pyfunc
class _PythonCallable:
"""

11
mindspore/dataset/engine/validators.py
View File

@ -742,14 +742,9 @@ def check_device_send(method):
@wraps(method)
def new_method(self, *args, **kwargs):
param, param_dict = parse_user_args(method, *args, **kwargs)
para_list = list(param_dict.keys())
if "prefetch_size" in para_list:
if param[0] is not None:
check_pos_int32(param[0], "prefetch_size")
type_check(param[1], (bool,), "send_epoch_end")
else:
type_check(param[0], (bool,), "send_epoch_end")
[send_epoch_end, create_data_info_queue], _ = parse_user_args(method, *args, **kwargs)
type_check(send_epoch_end, (bool,), "send_epoch_end")
type_check(create_data_info_queue, (bool,), "create_data_info_queue")
return method(self, *args, **kwargs)