@CaptainEven , Thanks for the nice repo. Have you used validation during training? If yes, where it is in your code?