                            Permute, Reshape, SeparableConv2D, Softmax, UpSampling2D)
  from keras.layers.advanced_activations import LeakyReLU
  from keras.layers.convolutional import Conv2D
  from keras.layers.core import Activation
  from keras.initializers import he_uniform, VarianceScaling
  from .initializers import ICNR, ConvolutionAware
  from .layers import PixelShuffler, SubPixelUpscaling, ReflectionPadding2D, Scale
  from .normalization import GroupNormalization, InstanceNormalization
  
  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
 
</s> remove         original = kwargs["kernel_initializer"]
 </s> add         original = kwargs.get("kernel_initializer", None) </s> remove         self.switch_kernel_initializer(kwargs, original_init)
 </s> add  </s> remove         original_init = self.switch_kernel_initializer(kwargs, VarianceScaling(
            scale=0.2,
            mode="fan_in",
            distribution="uniform"))
 </s> add  </s> remove                             force_initializer=True,
 </s> add 
from keras.initializers import he_uniform
                            Permute, Reshape, SeparableConv2D, Softmax, UpSampling2D)
  from keras.layers.advanced_activations import LeakyReLU
  from keras.layers.convolutional import Conv2D
  from keras.layers.core import Activation
from keras.initializers import he_uniform
  from .initializers import ICNR, ConvolutionAware
  from .layers import PixelShuffler, SubPixelUpscaling, ReflectionPadding2D, Scale
  from .normalization import GroupNormalization, InstanceNormalization
  
  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
keep keep keep keep replace keep keep keep keep keep
 Remove he_uniform from init from res block and use FS default 
lib/model/nn_blocks.py