From 9334ccd9007b28315bad2122a7f1f8caf8bbaea3 Mon Sep 17 00:00:00 2001 From: Sean Date: Fri, 20 Aug 2021 11:57:13 -0700 Subject: [PATCH] fixed stride typo --- 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 cce81694..9a32b270 100644 --- a/pytorchvideo/models/vision_transformers.py +++ b/pytorchvideo/models/vision_transformers.py @@ -237,12 +237,12 @@ def create_multiscale_vision_transformers( ) input_dims = [temporal_size, spatial_size, spatial_size] - input_stirde = ( + input_stride = ( (1,) + conv_patch_embed_stride if use_2d_patch else conv_patch_embed_stride ) patch_embed_dims = [ - 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)) ] cls_positional_encoding = SpatioTemporalClsPositionalEncoding( embed_dim=patch_embed_dim,