Hi,
For a university project, we have to create a certain accuracy level for the famous Fashion MNIST with model of neural network.
I am having some issues to somehow connect the test version to the train version. I am trying to save a the best version to then load it again to evaluate:
I am trying to do this code but I can’t get it right.
model = FashionMNISTNet(*images, **labels)
model.load_state_dict(torch.load(os.path.join))
model.eval()
TypeError                                 Traceback (most recent call last)
 in ()
----> 1 model = FashionMNISTNet(*images, **labels)
2 model.load_state_dict(torch.load(os.path.join))
3 model.eval()
TypeError: type object argument after ** must be a mapping, not Tensor