Error in conversion from Torch to TF model

[nfaraji2002]

Hi
I have been using convert_tacotron2_torch_to_tf.py for conversion of a downloaded Tacotron model to tf version, but I faced with an error:

AssertionError: [!] weight shapes does not match: decoder/while/attention/query_layer/linear_layer/kernel:0 vs decoder.attention.query_layer.weight --> (1024, 128) vs (128, 1024)

I think it is the bug of conversion code. Would you please help me to solve the issue?

Neda