Skip to content

Instantly share code, notes, and snippets.

@madratman
Created October 31, 2017 20:47
Show Gist options
  • Save madratman/d1e3efa2c28254be4f257381206c676c to your computer and use it in GitHub Desktop.
Save madratman/d1e3efa2c28254be4f257381206c676c to your computer and use it in GitHub Desktop.
import torch
import pytorch_autoencoder
list_of_numbers = get them from cpp
sum = sum
model = pytorch_autoencoder.ConvCVAENet(state_dim = 2,
condition_dim = (200, 200, 2),
latent_dim = 3,
encoder_hidden_dim = 512,
decoder_hidden_dim = 512)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment