Pytorch: Add normalization layer at the begining of a pre-trained model

Question

I have a pretrained UNet model with the following architecture

UNet(
  (encoder1): Sequential(
    (enc1conv1): Conv2d(3, 32, kernel_size=(3, 3), stride=(1, 1