Possible Bug in UNet Implementation

openai / guided-diffusion

MIT License

5.9k stars 782 forks source link

Possible Bug in UNet Implementation #140

Closed RoganInglis closed 4 months ago

RoganInglis commented 4 months ago

I have noticed that an extra set of ResBlock + optional AttentionBlock layers are added in the decoder compared to the encoder in the UNet implementation. In the paper it mentions that the encoder and decoder are mirrors of each other. Is this a bug?