facenet-pytorch
facenet-pytorch copied to clipboard
Load models offline, from model
How does one load the models from a folder?
Reviewing https://github.com/timesler/facenet-pytorch/blob/master/models/mtcnn.py does not indicate a user set for the file path of the model.
Any pointers appreciated, thank you.
I read the source code of InceptionResnetV1 and I figured out how to do that for InceptionResnetV1:
self.net = InceptionResnetV1(classify=True, num_classes=8631)
self.net.load_state_dict(torch.load(r'./media/checkpoints/20180402-114759-vggface2.pt'))