search

yang-song / score_sde_pytorch

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
https://arxiv.org/abs/2011.13456
Apache License 2.0
1.58k stars 295 forks source link

ConditionalResidualBlock not working #48

Open shuohantao opened 8 months ago

shuohantao commented 8 months ago

https://github.com/yang-song/score_sde_pytorch/blob/cb1f359f4aadf0ff9a5e122fe8fffc9451fd6e44/models/layers.py#L406-L416 The above code would create mismatched shortcut and output shape when dilation is larger than 1.