Skip to content

Instantly share code, notes, and snippets.

@SannaPersson

SannaPersson/0cc73738-bee4-4fe3-b97a-cc1a24e8acbd.py

Created September 13, 2022 13:53
Show Gist options
  • Save SannaPersson/f269c5baecdf53dcfadc0c524999084b to your computer and use it in GitHub Desktop.
Save SannaPersson/f269c5baecdf53dcfadc0c524999084b to your computer and use it in GitHub Desktop.
Download ZIP
variational_autoencoder4
Raw
0cc73738-bee4-4fe3-b97a-cc1a24e8acbd.py
# Initialize model, optimizer, loss
model = VariationalAutoEncoder(INPUT_DIM, Z_DIM).to(device)
optimizer = torch.optim.Adam(model.parameters(), lr=LR_RATE)
loss_fn = nn.BCELoss(reduction="sum")
# Run training
train(NUM_EPOCHS, model, optimizer, loss_fn)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment