From d9750b2bfec439dbc90c9476b219c2872d117a02 Mon Sep 17 00:00:00 2001 From: kajc10 Date: Wed, 21 Feb 2024 19:34:25 +0000 Subject: [PATCH] removed hardcoded num_samples in ddpm --- tools/sample_ddpm_vqvae.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tools/sample_ddpm_vqvae.py b/tools/sample_ddpm_vqvae.py index ac66245..3d71d8e 100644 --- a/tools/sample_ddpm_vqvae.py +++ b/tools/sample_ddpm_vqvae.py @@ -24,10 +24,7 @@ def sample(model, scheduler, train_config, diffusion_model_config, autoencoder_model_config['z_channels'], im_size, im_size)).to(device) - xt = torch.randn((2, - autoencoder_model_config['z_channels'], - im_size, - im_size)).to(device) + save_count = 0 for i in tqdm(reversed(range(diffusion_config['num_timesteps']))): # Get prediction of noise