Skip to content

Instantly share code, notes, and snippets.

@SannaPersson

SannaPersson/94878c54-1a7a-4792-ac44-f4ac81eed364.py

Created September 13, 2022 13:53
Show Gist options
  • Save SannaPersson/1f2c4db89a6866bbf32b73508db81458 to your computer and use it in GitHub Desktop.
Save SannaPersson/1f2c4db89a6866bbf32b73508db81458 to your computer and use it in GitHub Desktop.
Download ZIP
variational_autoencoder2
Raw
94878c54-1a7a-4792-ac44-f4ac81eed364.py
# Configuration
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
INPUT_DIM = 784
Z_DIM = 20
H_DIM = 200
NUM_EPOCHS = 10
BATCH_SIZE = 32
LR_RATE = 3e-4
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment