Difference Between Train Test Split Pytorch Tutorial