Browse Source

fix generatordataset check shuffle parameter

tags/v0.3.0-alpha
panfengfeng 6 years ago
parent
commit
5e7cef7e3d
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

@@ -570,6 +570,8 @@ def check_generatordataset(method):
check_param_type(nreq_param_int, param_dict, int)
nreq_param_list = ["column_types"]
check_param_type(nreq_param_list, param_dict, list)
nreq_param_bool = ["shuffle"]
check_param_type(nreq_param_bool, param_dict, bool)

num_shards = param_dict.get("num_shards")
shard_id = param_dict.get("shard_id")


Loading…
Cancel
Save