How to create 3d model of HRnet w48 in pytorch

Hi I am new to pytorch and I am wondering if there is a way to create HRnet for 3d image data for image reconstruction?

This topic migth be interesting as it discusses a similar use case.
However, I don’t know what the author ended up using.