From d75745bce073253ea338c22b3825f2e991255d4f Mon Sep 17 00:00:00 2001 From: Yanjun Peng Date: Thu, 2 Apr 2020 16:37:39 +0800 Subject: [PATCH] fix random crop parameter check --- mindspore/dataset/transforms/vision/validators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/dataset/transforms/vision/validators.py b/mindspore/dataset/transforms/vision/validators.py index 443232d78..caab120af 100644 --- 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: -- GitLab