How to replace a pre-trained BN layer with the SFT layer at the beginning of training

Hi xintao, I'm doing a SR working inspired by your SFTGAN. I would like to know how you replace a  pre-trained BN layer with the SFT layer when you start training? How to initialize the convolution layer which outputs gamma or beta? Thanks.

[https://github.com/xinntao/SFTGAN/issues/3#issuecomment-393222418](url)