- Added comprehensive inline comments explaining each step of the training process in the `run_nhrepnet_training` method
- Improved code structure by adding descriptive comments for variable initializations and key computational steps
- Enhanced code readability by breaking down complex operations with clear explanatory comments
- Maintained existing functionality while providing better code documentation
- 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
- Split the initialization process into multiple private methods for better readability and maintainability.
- Added detailed logging for each step of the initialization process, including error handling for missing parameters and file loading issues.
- Enhanced configuration and directory setup with clearer error messages and structured logging.
- Improved data loading methods to handle both single and list-based data inputs more robustly.
- Introduced methods for setting up the CSG tree and computing local sigma values, with appropriate logging for each operation.