!110 Fix RandomCrop parameter check

Merge pull request !110 from pengyanjun/fix_dataset_parameter_check_problem
2020-04-02 21:49:02 +08:00 · 2020-04-02 21:49:02 +08:00 · 1ace66c87c
parent b8bfaeb65a 49a41bb2cc
commit 1ace66c87c
1 changed files with 1 additions and 1 deletions
--- a/mindspore/dataset/transforms/vision/validators.py
+++ b/mindspore/dataset/transforms/vision/validators.py
@ -325,7 +325,7 @@ def check_random_crop(method):

    @wraps(method)
    def new_method(self, *args, **kwargs):
-        args = (list(args) + 4 * [None])[:5]
+        args = (list(args) + 5 * [None])[:5]
        size, padding, pad_if_needed, fill_value, padding_mode = args

        if "size" in kwargs: