Searched refs:num_batch (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_sync_wait.py | 96 dataset = dataset.sync_wait(num_batch=2, condition_name="every 2 batches")
193 dataset.sync_wait(num_batch=2, condition_name="every batch")
208 dataset.sync_wait(condition_name="every batch", num_batch=-1, callback=aug.update)
229 dataset.sync_update(condition_name="every batch", num_batch=-1, data=data)
262 dataset = dataset.sync_wait(condition_name='', num_batch=1)
284 dataset.sync_wait(condition_name="every batch", num_batch="123", callback=aug.update)
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 500 def sync_wait(self, condition_name, num_batch=1, callback=None): argument
545 return SyncWaitDataset(self, condition_name, num_batch, callback)
1764 def sync_update(self, condition_name, num_batch=None, data=None): argument
1775 if (not isinstance(num_batch, int) and num_batch is not None) or \
1776 (isinstance(num_batch, int) and num_batch <= 0):
1779 …se RuntimeError("Sync_update batch size can only be positive integer, got : {}.".format(num_batch))
1788 if num_batch is not None:
1789 num_batch *= self.get_batch_size()
1790 notifiers_dict[condition_name](num_batch, data)
2408 def __init__(self, input_dataset, condition_name, num_batch, callback=None): argument
[all …]
|
| D | validators.py | 656 [condition_name, num_batch, _], _ = parse_user_args(method, *args, **kwargs)
659 type_check(num_batch, (int,), "num_batch")
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | yolo.py | 200 num_batch = P.Shape()(x)[0]
204 prediction = P.Reshape()(x, (num_batch,
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | yolov3.py | 419 num_batch = P.Shape()(x)[0]
423 prediction = P.Reshape()(x, (num_batch,
|