Browse Source

!689 Check num_shards and shard_id in TextFileDataset

Merge pull request !689 from yanghaitao/yht_tffileddataset_shard
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
86075dc9cb
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/dataset/engine/validators.py

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

@@ -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

Loading…
Cancel
Save