yl-1993 / learn-to-cluster

Learning to Cluster Faces (CVPR 2019, CVPR 2020)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Pretrained Face Recognition Model in MS1M datasets

houguanqun opened this issue · comments

When I decompressed the file ’resnet50_part0_train.pth.tar‘, it was found to be damaged, can you upload it again?

Hi @houguanqun , the filename maybe a little confusing as it is not a tar, but a model itself. Would you mind trying the following code?

import torch
ckpt = torch.load('resnet50_part0_train.pth.tar')
print(ckpt['state_dict'].keys())

Thank you very much!