tamerthamoqa / facenet-pytorch-glint360k

A PyTorch implementation of the 'FaceNet' paper for training a facial recognition model with Triplet Loss using the glint360k dataset. A pre-trained model using Triplet Loss is available for download.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

torch.load pre-trained model error

2017234312 opened this issue · comments

hello, thank you for your contribution. i download your weight file and load it to check the performance on LFW , while load the pt file according to your description, the error "_pickle.UnpicklingError: A load persistent id instruction was encountered,
but no persistent_load function was specified." occurred. the pickle and save operation for your weight file maybe error???

Hello 2017234312,

I have managed to recreate this issue by using a Python 2.7 conda environment. I'm afraid you would have to use Python 3.6+ in order to use PyTorch 1.6. I have also updated the README on importing the required files in a more correct way.

I hope this was helpful, please inform me if other issues arise.