I added the PSA_p module on BasicBlock as the segmentation code did here. However, it has no effect on the final results. The final results are even 0.1% less than directly using the HRNet v2. I used the HRNet pose estimation default training schedule.