class VAE(nn.Module): def __init__(self): super(VAE, self).__init__() self.fc1 = nn.Linear(784, 400) model = VAE() with open('model_save', 'wb') as f: torch.save(model, f)
When I open
model_save file, I see the following: mode_save is not UTF encoded. Saving Disabled. See console for more details. And, When I try to use torch.load, I see this error: "
module" object has no attribute VAE. Would someone please advise me? Thanks.