fix random crop parameter check

This commit is contained in:
Yanjun Peng 2020-04-02 16:37:39 +08:00
parent 9cb219c683
commit 49a41bb2cc
1 changed files with 1 additions and 1 deletions

View File

@ -325,7 +325,7 @@ def check_random_crop(method):
@wraps(method) @wraps(method)
def new_method(self, *args, **kwargs): 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 size, padding, pad_if_needed, fill_value, padding_mode = args
if "size" in kwargs: if "size" in kwargs: