U-Net from Scratch using PyTorch | Backbone for Diffusion Models

[@tamineabderrahmane248]

i got you , you are maxime

[@papersin100linesofcode]

Indeed :)

[@rahulachu5747]

In the get_timestep_embedding, for padding shouldn’t we use torch.nn.functional.pad(emb,[0,1,0,0]). I think the syntax is from tensor flow padding. But thanks a lot for this amazing tutorial.

[@papersin100linesofcode]

You are absolutely right, thank you for the catch! This function was indeed borrowed from tensorflow. Feel free to make a pull request. Otherwise, I will update it myself.