Pytorch Data Format, The code to create the AG News model is from this PyTorch tutorial.