fix random crop parameter check

5 years ago · d75745bce0
parent f2263572fb
commit d75745bce0
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: