InvalidArgumentError when changing the batch size

santhoshkolloju / Abstractive-Summarization-With-Transfer-Learning

Abstractive summarisation using Bert as encoder and Transformer Decoder

406 stars 98 forks source link

Open astariul opened 5 years ago

astariul commented 5 years ago

I tried to run the code with a batch size of 8, however I got this error :

InvalidArgumentError (see above for traceback): Incompatible shapes at component 0: expected [1,512] but got [8,512].

maozezhong commented 5 years ago

Have you solved this issue? @Colanim

astariul commented 5 years ago

No 😕