From 4df1ca13ec22c57fc05c286470325bf3fbc4d4b9 Mon Sep 17 00:00:00 2001 From: Anirudh Thatipelli Date: Mon, 31 Jan 2022 17:41:01 +0530 Subject: [PATCH] Update vision_transformers.py Corrected spelling mistake in stride. --- pytorchvideo/models/vision_transformers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorchvideo/models/vision_transformers.py b/pytorchvideo/models/vision_transformers.py index 6482d96c..69b3e799 100644 --- a/pytorchvideo/models/vision_transformers.py +++ b/pytorchvideo/models/vision_transformers.py @@ -247,14 +247,14 @@ def create_multiscale_vision_transformers( ) input_dims = [temporal_size, spatial_size[0], spatial_size[1]] - input_stirde = ( + input_stride = ( (1,) + tuple(conv_patch_embed_stride) if use_2d_patch else conv_patch_embed_stride ) patch_embed_shape = ( - [input_dims[i] // input_stirde[i] for i in range(len(input_dims))] + [input_dims[i] // input_stride[i] for i in range(len(input_dims))] if enable_patch_embed else input_dims )