Is FashionMNISTNet a subclass of torch.nn.Module? If so it is a bit strange to pass images and labels to the __init__ function as usually the model definition itself doesn’t need to know anything about the data.
Can you show what the definition of FashionMNISTNet looks like?
The Imagenet Example also gives an example of how to use the model loading functions.