get nan result for a whole batch

peteryuX / arcface-tf2

ArcFace unofficial Implemented in Tensorflow 2.0+ (ResNet50, MobileNetV2). "ArcFace: Additive Angular Margin Loss for Deep Face Recognition" Published in CVPR 2019. With Colab.

MIT License

263 stars 60 forks source link

get nan result for a whole batch #29

Closed gitlabspy closed 3 years ago

gitlabspy commented 3 years ago

Hi, Thanks for sharing this amazing work! I downloaded your model and loaded the weights, resnet-50 ccrop=true, and I test it with a bunch of images. Some batches work fine, no nan, but some batches are all nan results. What might cause this?

sfxnminimah commented 3 years ago

Hi, Pls did you solve this issue? Getting the same nan results for some batches too