How to setup and import custom_attn?

[bigrat911]

I guess custom_attn is a customized package,
so how can I setup and import it, Thanks!

[sujitpal]

Right now it's not a true Python package, if you copy it to your current directory and put "import custom_attn", then you should be able to refer to the custom layers in it with custom_attn.AttentionM (for example).