Skip to content

Instantly share code, notes, and snippets.

@monajalal

monajalal/VAE_training.ipynb

Last active November 12, 2020 00:05
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save monajalal/8f0226aba24c98235667a5f0a1216aca to your computer and use it in GitHub Desktop.
Save monajalal/8f0226aba24c98235667a5f0a1216aca to your computer and use it in GitHub Desktop.
Download ZIP
trying a VAE training on my custom images
Display the source blob
Display the rendered blob
Raw
VAE_training.ipynb
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@monajalal
Copy link
Author

Changing the my_transform to the following didn't solve the problem:

my_transform = transforms.Compose([
    transforms.ToPILImage(),
    transforms.ToTensor(),
    transforms.Resize((100,100))
                      ])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment