Activation function in UBlock
hsiaohan0827 opened this issue · comments
Hi, I noticed that each downsampling stage resembles a DWConv+LN+PReLU, as shown in algorithm 1 in the paper.
But it seems like the code of this downsampling process contains only DWConv+LN in Line 147.
Is PReLU applied in the UBlock? Thanks.
You are right, this was probably a typo in the paper, I do not think that makes a difference besides increasing the parameters a bit.