Skip to content

Commit 88ea2c6

Browse files
author
zyan3
committed
[video transforms] fix a bug
1 parent cf66708 commit 88ea2c6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/transforms/transforms_video.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def __repr__(self):
8282
class CenterCropVideo(object):
8383
def __init__(self, crop_size):
8484
if isinstance(crop_size, numbers.Number):
85-
self.crop_size = (int(size), int(size))
85+
self.crop_size = (int(crop_size), int(crop_size))
8686
else:
8787
self.crop_size = crop_size
8888

0 commit comments

Comments
 (0)