Skip to content

Possible bug in the padding mask handling #14

Description

@bjourne

I've stared at these lines in your excellent tutorial for a while now:

 enc_padding_mask = tf.keras.layers.Lambda(
     create_padding_mask, output_shape=(1, 1, None),
     name='enc_padding_mask')(inputs)
 # mask the future tokens for decoder inputs at the 1st attention block
 look_ahead_mask = tf.keras.layers.Lambda(
     create_look_ahead_mask,
     output_shape=(1, None, None),
     name='look_ahead_mask')(dec_inputs)
 # mask the encoder outputs for the 2nd attention block
 dec_padding_mask = tf.keras.layers.Lambda(
     create_padding_mask, output_shape=(1, 1, None),
     name='dec_padding_mask')(inputs)

enc_padding_mask and dec_padding_mask will always be equal. Is this intentional? It seems weird to create two different padding masks that are the same.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions