Skip to content

Instantly share code, notes, and snippets.

@williamFalcon

williamFalcon/ae.py

Created February 7, 2020 11:42
Show Gist options
  • Save williamFalcon/083ebcd7e7887c4441de406097cbadfb to your computer and use it in GitHub Desktop.
Save williamFalcon/083ebcd7e7887c4441de406097cbadfb to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ae.py
# models
encoder = Encoder()
decoder = Decoder()
# data are images
image = data.x
# encode and decode
z = encoder(image)
x_hat = decoder(z)
# mean square error
loss = MSE(x, x_hat)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment