Check TfFileDataset num_shards and shard_id

60f26633 · yanghaitao · 7c7d95ac · 60f26633
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

mindspore/dataset/engine/validators.py mindspore/dataset/engine/validators.py +2 -0

未找到文件。
--- a/mindspore/dataset/engine/validators.py
+++ b/mindspore/dataset/engine/validators.py
@@ -906,6 +906,8 @@ def check_textfiledataset(method):

        check_param_type(nreq_param_int, param_dict, int)

+        check_sampler_shuffle_shard_options(param_dict)
+
        return method(*args, **kwargs)

    return new_method