https://ideas.creativecasinoties.com/post/using-dropout-regularization-in-pytorch-models-3k8zif