提交 d75745bc 编写于 作者: Y Yanjun Peng 提交者: 高东海

fix random crop parameter check

上级 f2263572
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册