Model works fine during training, but same output on Validation Set

https://www.kaggle.com/shubhendumishra/recognizing-faces-in-the-wild-vggface-pytorch/

Here is a kernel I’m working on. Can somebody tell me why this is happening? During training model outputs are fine but during Validation only one class is the output