55d1c4d1460ed72750c51f7e1c057e0202d175e3,tests/layers/test_transformer.py,TestMultiHead,test_fit,#TestMultiHead#,35

Before Change

            name="Input",
        )
        dense_layer = keras.layers.Dense(units=3, name="Dense-1")(input_layer)
        transformer_layer = get_transformer(
            inputs=dense_layer,
            head_num=3,
            hidden_dim=12,
            dropout=0.001,
            name="Transformer-1",
        )
        transformer_layer = get_transformer(
            inputs=transformer_layer,
            head_num=3,
            hidden_dim=12,

After Change

            dropout_rate=0.001,
            name="Transformer-1",
        )(dense_layer)
        transformer_layer = Transformer(
            head_num=3,
            hidden_dim=12,
            dropout_rate=0.001,
            name="Transformer-2",
        )(transformer_layer)
        dense_layer = keras.layers.Dense(units=3, name="Dense-2")(transformer_layer)
        model = keras.models.Model(
            inputs=input_layer,
            outputs=dense_layer,

Instances