1, 28, 28).detach().numpy() print("Latent Space of VAE: ") Util.printGridMNIST(latent_space_VAE, n, save_name='MNIST_Latent_Space_VAE') print("Latent Space of IWAE: ") Util.printGridMNIST(latent_space_IWAE, n, save_name='MNIST_Latent_Space_IWAE') """ Print dimensionality reduction examples WARNING: Only works if dimension of z is 2 """ print("Dimensionality Reduction of VAE: ") Util.MNIST_2d_reduction(VAE_model_2D, TestDataset, save_name='MNIST_2d_reduction_VAE') print("Dimensionality Reduction of IWAE: ") Util.MNIST_2d_reduction(IWAE_model_2D, TestDataset, save_name='MNIST_2d_reduction_IWAE') """ Calculate Negative Log-Likelihood in 2D WARNING: Need to set Settings.device = 'cpu' and re-import Settings.py """ from Settings import Settings device = torch.device('cpu') #importe relevant model