def visualize(ss_vae, viz, test_loader): if viz: plot_conditional_samples_ssvae(ss_vae, viz) mnist_test_tsne_ssvae(ssvae=ss_vae, test_loader=test_loader)
def visualize(ss_vae, viz, test_loader): if viz is not None: plot_conditional_samples_ssvae(ss_vae, viz) mnist_test_tsne_ssvae(ssvae=ss_vae, test_loader=test_loader)