Load Data and Train Model - Deep Learning with PyTorch 6