Fix manual merge for ViT

Summary: My manual cherrypick for https://www.internalfb.com/intern/pick/request/permalink/1290074624677854/ introduced errors which I'm fixing here.

Reviewed By: lauragustafson

Differential Revision: D24996390

fbshipit-source-id: de7b7227f055ce3a491ae55a2fe094afa6d424f6

classy_vision/models/vision_transformer.py

@@ -13,13 +13,15 @@
 """
 
 import copy
+import logging
 import math
 from collections import OrderedDict
 from functools import partial
 
 import torch
 import torch.nn as nn
 from classy_vision.models import ClassyModel, register_model
+
 from .lecun_normal_init import lecun_normal_init
 
 
@@ -143,7 +145,7 @@ def __init__(
         classifier="token",
     ):
         super().__init__()
-        assert image_size % patch_size == 0, "Input shape indivisble by patch size"
+        assert image_size % patch_size == 0, "Input shape indivisible by patch size"
         assert classifier in ["token", "gap"], "Unexpected classifier mode"
         self.image_size = image_size
         self.patch_size = patch_size
@@ -178,6 +180,7 @@ def __init__(
         )
         self.trunk_output = nn.Identity()
 
+        self.seq_length = seq_length
         self.init_weights()
 
     def init_weights(self):