03f07734648aa9843393f0ebf19c61c2a34c728e,cleverhans/utils_keras.py,KerasModelWrapper,fprop,#KerasModelWrapper#,182

Before Change

      new_input = self.model.get_input_at(0)

      // Make a new model that returns each of the layers as output
      outnodes_last = [x_layer.get_output_at(-1) for x_layer in self.model.layers]
      outnodes_first = [x_layer.get_output_at(0) for x_layer in self.model.layers]
      if outnodes_last != outnodes_first:
        warnings.warn("Mutiple output nodes detected, picking last ones as default."
                      "This could happen due to using of stacked model.")

After Change

        warnings.warn("Abstract layer detected, picking last ouput node as default."
                      "This could happen due to using of stacked model.")
    
      layer_outputs = []
      // For those abstract model layers, return their last output node as default.
      for x_layer in self.model.layers:
        if x_layer.name not in abstract_layers:
          layer_outputs.append(x_layer.output)
        else:
          layer_outputs.append(x_layer.get_output_at(-1))
          
      self.keras_model = KerasModel(new_input, layer_outputs)

    // and get the outputs for that model on the input x
    outputs = self.keras_model(x)

Instances