Virgil
Results
1
comments of
Virgil
`DownSample` use `Conv2d` with kernel_size `(3, 3)`, but `UpSample` use `ConvTranspose2d` with kernel_size `(4, 4)` which results in the `(H, W)` of `x` larger than `s`. I add a `F.pad`...