santhoshkolloju / Abstractive-Summarization-With-Transfer-Learning

Abstractive summarisation using Bert as encoder and Transformer Decoder
406 stars 98 forks source link

got an unexpected keyword argument 'embedding' #46

Open noabenefraim opened 4 years ago

noabenefraim commented 4 years ago

In model.py line 114

decoder = tx.tf.modules.TransformerDecoder(embedding=tgt_embedding,
                             hparams=dcoder_config)

I am getting a type error that there is an unexpected keyword embedding passed into the TransformerDecoder. How did people resolve this? I see that the Transformer Decoder takes in (vocab_size=None, output_layer=None, hparams=None). So I'm not sure what the embedding refers to here.

Any guidance would be appreciated.