Data is handled with raw Python lists. Much more convenient and efficient to use PyTorch Dataset instead.