No module named 'wSEAM' when import wSEAM.network.resnet38d in resnet38_aff.py

[ubun23]

Thanks for your outstanding work on weak segmentation! But when I reproduced this code in step2: Refine with AffinityNet, AffinityNet train, an error occurred. That is No module named 'wSEAM' when "import wSEAM.network.resnet38d in resnet38_aff.py". I checked this code carefully, but I didn't find any code about wSEAM. I would be grateful if anyone could help me solve it!!