!110 Fix RandomCrop parameter check
Merge pull request !110 from pengyanjun/fix_dataset_parameter_check_problem
This commit is contained in:
commit
1ace66c87c
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue