model = InceptionResnetV1(pretrained='vggface2')
checkpoint = torch.load("/content/drive/MyDrive/export/InceptionResNetV1_ArcFace.pt")
model.load_state_dict(checkpoint)
model = model.cuda()
`ModuleNotFoundError Traceback (most recent call last)
[<ipython-input-6-9840bef75f84>](https://localhost:8080/#) in <module>()
1 model = InceptionResnetV1(pretrained='vggface2')
----> 2 checkpoint= torch.load("/content/drive/MyDrive/export/InceptionResNetV1_ArcFace.pt")
3 model.load_state_dict(checkpoint)
4 model = model.cuda()
2 frames
[/usr/local/lib/python3.7/dist-packages/torch/serialization.py](https://localhost:8080/#) in find_class(self, mod_name, name)
873 def find_class(self, mod_name, name):
874 mod_name = load_module_mapping.get(mod_name, mod_name)
--> 875 return super().find_class(mod_name, name)
876
877 # Load the data (which may in turn use `persistent_load` to load tensors)
ModuleNotFoundError: No module named 'model'`
Can you please tell me what did I do wrong and how to fix it? I'm a beginner so I don't really know what to do here.