https://ideas.creativecasinoties.com/post/using-dropout-regularization-in-pytorch-models-997a7q