I have a model:
cnn=CNN()
torch.save(cnn, ‘./model/trained_model.pt’)
model = torch.load(’./model/trained_model.pt’)
AttributeError: Can’t get attribute ‘CNN’ on <module ‘main’>
why  am I getting this error and how to solve it?