I'm trying to load a previously trained model using the following:
model = torch.load(f"{SaveFolder}/model.pt", map_location = torch.device('cpu'))
SaveFolder has the correct path of the previously saved fully trained model. And this is the error I'm getting: