Pytorch Machine Learning Models And Types