denoising-diffusion-pytorch icon indicating copy to clipboard operation
denoising-diffusion-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon lucidrains

Final ResNetBlock block for UNet in denoising_diffusion_pytorch.py

Open MekkCyber opened this issue 2 years ago • 0 comments

Hello @lucidrains

Sorry if i'm mistaken, but in the line 380 of the file denoising_diffusion_pytorch.py, where the final ResNetBlock is defined for the UNet : self.final_res_block = block_klass(dim * 2, dim, time_emb_dim = time_dim) Shouldn't the input channels dimension be init_dim * 2 instead of dim * 2 ?

MekkCyber avatar Oct 27 '23 14:10 MekkCyber