fix random crop parameter check
This commit is contained in:
parent
9cb219c683
commit
49a41bb2cc
|
@ -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:
|
||||||
|
|
Loading…
Reference in New Issue