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.