fix random crop parameter check

2020-04-02 16:37:39 +08:00 · 2020-04-02 16:37:39 +08:00 · 49a41bb2cc
parent 9cb219c683
commit 49a41bb2cc
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: