Honee-W/CPTNN

Downsampler

Mohammadhakimkhah opened this issue · 2 comments

Capture
Capture1

in Downsampler module, shouldn't you put "Layer-norm" and "PRelu" after 2-D conv ?

image
image

in Upsampler module too, shouldn't put "Layer-norm" and "PRelu" after subpixel convolution ?

Thanks for the feedback. The code has been updated.