Skip to content

some typo #5

@jszhujun2010

Description

@jszhujun2010

learned a lot from this, some typo found either.
line 177, shape should be [batch, seq_len, target_vocab]

# Make the prediction out of the decoder output.<br/>
logits = tf.layers.dense(dec_out, target_vocab)  # [batch, target_vocab]

line 433, should be num_dec_layers

for i in range(self.num_enc_layers):
    out = self.decoder_layer(out, enc_out, input_mask, target_mask, f'dec_{i}')

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