How to use MNIST model

vision
4

If you’re loading the images using PIL, you can use torchvision.transforms.Resize.
As you can see in the example, you can compose several transformations using torchvision.transforms.Compose.