`test_data = torchvision.datasets.FashionMNIST("./", train=True, transform=transforms.ToTensor(), download=True)` should use `train=False` This changes the figures in this chapter significantly.