How to train on MNIST or FashionMNIST

[rualz]

Hi, how should I modify the code to train the fashionmnist or mnist data set, their size is 28x28, which is not a supported size, how to modify the code if I want to train them, like if image_size == 32, how to set the channel_mult. Thanks