751eaead545d45bcc47bff7d82656b08c474b434,privacy/bolton/model.py,Bolton,compile,#Bolton#Any#Any#Any#Any#Any#Any#Any#Any#,104

Before Change

    See super class. Default optimizer used in Bolton method is SGD.

    
    if not isinstance(loss, StrongConvexLoss):
      raise ValueError("Loss must be subclassed from StrongConvexLoss")
    self.output_layer.kernel_regularizer.l2 = loss.reg_lambda()
    if not isinstance(optimizer, Private):
      optimizer = optimizers.get(optimizer)
      if isinstance(self.optimizer, trackable.Trackable):

After Change

    if not self._layers_instantiated:  // compile may be called multiple times
      kernel_intiializer = kwargs.get("kernel_initializer",
                                      tf.initializers.GlorotUniform)
      self.output_layer = tf.keras.layers.Dense(
          self.n_classes,
          kernel_regularizer=loss.kernel_regularizer(),
          kernel_initializer=kernel_intiializer(),
      )
      self._layers_instantiated = True
    self.output_layer.kernel_regularizer.l2 = loss.reg_lambda
    if not isinstance(optimizer, Private):
      optimizer = optimizers.get(optimizer)

Instances