- Replaced print statements with logger.info() in ReconstructionRunner class - Added logging for input and output tensor shapes in NHRepNet forward method - Improved logging consistency and added docstring for network forward method