ZhaoJ9014 / face.evoLVe

🔥🔥High-Performance Face Recognition Library on PaddlePaddle & PyTorch🔥🔥
MIT License
3.44k stars 760 forks source link

amsoftmax the size not match? #48

Closed mmxuan18 closed 5 years ago

mmxuan18 commented 5 years ago
cos_theta = torch.mm(embbedings, kernel_norm)

RuntimeError: size mismatch, m1: [1 x 5994], m2: [512 x 5994]

i use a resnext as backbone and pass the output feature to amsoftmax, but error as upon, the kernel size type not match the feature type, something like the ordre is not correctly, am i used not correct?