Raw2Rgb network doesnt match the one in paper

[dongrongliang]

In the forward method of Raw2Rgb module in cycleisp.py, x = self.body【-1】(x) indicates that x is activated by 'nn.PReLU(n_feats)'. As the Figure-2 of paper show, it should be followed by the k-th RRG block here instead of activtion.

def forward(self, x, ccm_feat):
    x = self.head(x)
    for i in range(len(self.body)-1):
        x = self.body[i](x)
    body_out = x.clone()
    x = x * ccm_feat          ## Attention 
    x = x + body_out       
    x = self.body[-1](x)
    x = self.tail(x)
    x = self.tail_rgb(x)
    x = nn.functional.pixel_shuffle(x, 2)
    return x