search

SashaMalysheva / Pytorch-VAE

This is an implementation of the VAE (Variational Autoencoder) for Cifar10
MIT License
64 stars 21 forks source link

fixed IndexError #1

Closed pymarkus closed 2 years ago

pymarkus commented 2 years ago

Corrected IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

As of 20-Dec-2021 the code runs and model trains.