15a302edf591d62c7cc3353cb5b08625087ce548,official/vision/beta/projects/yolo/modeling/layers/nn_blocks.py,DarkResidual,build,#DarkResidual#Any#,271

Before Change

                           leaky_alpha=self._leaky_alpha)

    self._shortcut = ks.layers.Add()
    self._activation_fn = ks.layers.Activation(activation=self._sc_activation)

    super().build(input_shape)

  def call(self, inputs):

After Change

    // self._activation_fn = ks.layers.Activation(activation=self._sc_activation)
    if self._sc_activation == "leaky":
      alpha = {"alpha": self._leaky_alpha}
      self._activation_fn = partial(tf.nn.leaky_relu, **alpha)
    elif self._sc_activation == "mish":
      self._activation_fn = lambda x: x * tf.math.tanh(tf.math.softplus(x))
    else:
      self._activation_fn = tf_utils.get_activation(self._sc_activation)

Instances