init model
Init Linearly decoded VAE model object with given hyperparameters.
Parameters
device: str Device to run the training on (CPU or Cuda if GPU is available).
epochs: str Number of epochs to train models for.
learning_rate: float Adjustable learning rate for improving gradient descent opttimizer.
Web view
