Train for 3 channels issue

[SavvasMel]

Hello,

Thank you for uploading the training codes of the paper, and well done for you work!

I am trying to training the RealSN-DnCNN on coloured images (3 channels) with a Lips of 1 but I can see that the codes are not adapted for that. I made some changes myself for example I added

    if module.weight.shape[0] == 3:
        C_out = 3

The NN seems to be trained however when I print the spectral values σ(Κ) after a point they go much higher than 1. This leads ADMM to a strange behaviour even if the NN is denoising really well.

It seems that by introducing the 3 channels, there is a bag in the code that it gets activated but not when you require only 1 channel. Could I kindly ask what changes would you make to the codes to be trained on coloured images? It would be so helpful.

Thank you very much again,
Savvas