How to use PSA in human pose estimation？

It's my honor to see such an excellent job.

I use hrnetv2.py in the network to train MPIIi dataset, but the effect has not changed. How do I use the PSA module? 

Look forward to your human pose estimation code.Thank you!