Browse Source

Pre Merge pull request !16144 from shenwei41/sw_master

pull/16144/MERGE
shenwei41 Gitee 5 years ago
parent
commit
3d83cbbcc1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/dataset/core/validator_helpers.py

+ 1
- 1
mindspore/dataset/core/validator_helpers.py View File

@@ -334,7 +334,7 @@ def check_sampler_shuffle_shard_options(param_dict):
raise RuntimeError("sampler and num_samples cannot be specified at the same time.")

if num_shards is not None:
check_pos_int32(num_shards)
check_pos_int32(num_shards, "num_shards")
if shard_id is None:
raise RuntimeError("num_shards is specified and currently requires shard_id as well.")
check_value(shard_id, [0, num_shards - 1], "shard_id")


Loading…
Cancel
Save