Fixed bug

pytorch_pretrained_vit/__init__.py

@@ -1,5 +1,5 @@
-__version__ = "0.0.6"
+__version__ = "0.0.7"
 
 from .model import ViT
 from .configs import *
-from .utils import load_pretrained_weights
+from .utils import load_pretrained_weights

pytorch_pretrained_vit/utils.py

@@ -1,6 +1,7 @@
 """utils.py - Helper functions
 """
 
+import numpy as np
 import torch
 from torch.utils import model_zoo
 
@@ -107,6 +108,7 @@ def resize_positional_embedding_(posemb, posemb_new, has_class_token=True):
     zoom_factor = (gs_new / gs_old, gs_new / gs_old, 1)
     posemb_grid = zoom(posemb_grid, zoom_factor, order=1)
     posemb_grid = posemb_grid.reshape(1, gs_new * gs_new, -1)
+    posemb_grid = torch.from_numpy(posemb_grid)
 
     # Deal with class token and return
     posemb = torch.cat([posemb_tok, posemb_grid], dim=1)

setup.py

@@ -18,7 +18,7 @@
 EMAIL = 'luke.melas@gmail.com'
 AUTHOR = 'Luke'
 REQUIRES_PYTHON = '>=3.5.0'
-VERSION = '0.0.6'
+VERSION = '0.0.7'
 
 # What packages are required for this module to be executed?
 REQUIRED = [