How to use the attention function?

[314rated]

trafficstars

Hi, Thanks for the well documented and organised code! I have a few doubts. Firstly, on training the GAN on my own dataset, the output doesn't abide by the text. Also, there is heavy mode collapse. Could you suggest how to treat it? Also, could you please guide with an example on how to use the attention function provided? Thanks!