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
