We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cf66708 commit 88ea2c6Copy full SHA for 88ea2c6
torchvision/transforms/transforms_video.py
@@ -82,7 +82,7 @@ def __repr__(self):
82
class CenterCropVideo(object):
83
def __init__(self, crop_size):
84
if isinstance(crop_size, numbers.Number):
85
- self.crop_size = (int(size), int(size))
+ self.crop_size = (int(crop_size), int(crop_size))
86
else:
87
self.crop_size = crop_size
88
0 commit comments