ModuleNotFoundError: No module named 'models'

[elaine-sui]

Hi,

I encountered the error ModuleNotFoundError: No module named 'models' when loading the pretrained model checkpoint. Would it be possible to link another checkpoint version that is loadable without needing the code?

[elaine-sui]

Nevermind! Figured it out! Had to create a file video_classification.py in directory models that has the following:

import torch.nn as nn
import timm

class TIMMModel(nn.Module):
def init(self):
self.model = timm.create_model("vit_base_patch16_224", pretrained=True)