2443550ee6915daf6e7ff6306f3dc2922752ea4b,vis/backend/tensorflow_backend.py,,modify_model_backprop,#Any#Any#,41

Before Change

    // Restore original model to keep upstream callers unaffected.
    idx = 0
    for layer in model.layers[1:]:
        if hasattr(layer, "activation"):
            layer.activation = original_activations[idx]
            idx += 1

    // Register modifier.
    modifier_fn = _BACKPROP_MODIFIERS.get(backprop_modifier)
    if modifier_fn is None:
        raise ValueError(""{}" modifier is not supported".format(backprop_modifier))
    modifier_fn(backprop_modifier)

After Change

        for i, layer in utils.reverse_enumerate(modified_model.layers):
            if hasattr(layer, "activation"):
                layer.activation = tf.nn.relu
            if isinstance(layer, _ADVANCED_ACTIVATIONS):
                // NOTE: This code is brittle as it makes use of Keras internal serialization knowledge and might
                // break in the future.
                modified_layer = Activation("relu")
                modified_layer.inbound_nodes = layer.inbound_nodes
                modified_layer.name = layer.name
                modified_model.layers[i] = modified_layer

        // 3. Save model with modifications.
        modified_model.save(model_path)

        // 4. Register modifier and load modified model under custom context.
        modifier_fn = _BACKPROP_MODIFIERS.get(backprop_modifier)

Instances