reference https://github.com/jacobgil/pytorch-grad-cam https://github.com/davidsandberg/facenet/tree/master/src