Value Error

[annatn998]

Hi! I'm trying to use the PPVAE with my own dataset, which is just at .txt file of sentences each one on a new line:

ex:

sentence1 sentence2 sentence3

etc.

Everything runs ok up until this line the pretrainVAE.py code:

decoder_h_attn = Attention(head_num, head_size[layer], max_len)([decoder_h, decoder_h, decoder_h])

Then I get the error:

ValueError: Dimensions must be equal, but are 8 and 17 for 'attention_4/sub_1' (op: 'Sub') with input shapes: [?,8,8,?], [1,17,17,1].

I would really love to learn how to use this if you could help me.