diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index 3d8b5b2ed0..5504cc3362 100644 --- a/mindspore/dataset/engine/datasets.py +++ b/mindspore/dataset/engine/datasets.py @@ -208,7 +208,6 @@ class Dataset: Add a blocking condition to the input Dataset Args: - input_dataset (Dataset): Input dataset to apply flow control num_batch (int): the number of batches without blocking at the start of each epoch condition_name (str): The condition name that is used to toggle sending next row callback (function): The callback funciton that will be invoked when sync_update is called @@ -921,10 +920,13 @@ class Dataset: def sync_update(self, condition_name, num_batch=None, data=None): """ - condition_name (str): The condition name that is used to toggle sending next row - num_batch (int or None): The number of batches(rows) that are released - when pass_rows is None, will update the same number as sync_wait specified - data (dict or None): The data passed to the callback + Release a blocking condition and triger callback with given data + + Args: + condition_name (str): The condition name that is used to toggle sending next row + num_batch (int or None): The number of batches(rows) that are released + When num_batch is None, it will default to the number specified by the sync_wait operator + data (dict or None): The data passed to the callback """ notifiers_dict = self.get_sync_notifiers() if condition_name not in notifiers_dict: