I am training a deep learning model on the purpose of classifying images of playing cards using convolutional nueral networks. This is the my model , it’s architecture and training loop

I am currently implementing some neural networks for a image classification task, and i was wondering what the “best” approach of the following is.