Check TfFileDataset num_shards and shard_id

2020-04-26 15:10:41 +08:00 · 2020-04-26 15:10:41 +08:00 · 60f2663326
parent 7c7d95acf9
commit 60f2663326
1 changed files with 2 additions and 0 deletions
--- 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