Pytorch model is trained with 99% validation accuracy but during prediction predicts only one class

Question

I started to build an end-to-end face recognition system from facenet-pytorch library and I\'m having a problem with the prediction part. After the training process is done