Project-MONAI
PerceptualLoss Bug
Describe the bug When I use PerceptualLoss, if the loss is too low, it may occur "RuntimeError: Function ‘SqrtBackward0‘ returned nan values in its 0th output", I will provide the specific error location later.
Environment
MONAI version: 1.4.0
Numpy version: 1.26.4
Pytorch version: 2.4.0+cu118
MONAI flags: HAS_EXT = False, USE_COMPILED = False, USE_META_DICT = False
MONAI rev id: 46a5272196a6c2590ca2589029eed8e4d56ff008
MONAI file: /
Optional dependencies: Pytorch Ignite version: NOT INSTALLED or UNKNOWN VERSION. ITK version: NOT INSTALLED or UNKNOWN VERSION. Nibabel version: NOT INSTALLED or UNKNOWN VERSION. scikit-image version: NOT INSTALLED or UNKNOWN VERSION. scipy version: 1.15.2 Pillow version: 11.1.0 Tensorboard version: 2.19.0 gdown version: NOT INSTALLED or UNKNOWN VERSION. TorchVision version: 0.19.0+cu118 tqdm version: 4.67.1 lmdb version: NOT INSTALLED or UNKNOWN VERSION. psutil version: 7.0.0 pandas version: NOT INSTALLED or UNKNOWN VERSION. einops version: 0.8.1 transformers version: NOT INSTALLED or UNKNOWN VERSION. mlflow version: NOT INSTALLED or UNKNOWN VERSION. pynrrd version: NOT INSTALLED or UNKNOWN VERSION. clearml version: NOT INSTALLED or UNKNOWN VERSION.
Perhaps related to this issue? https://github.com/richzhang/PerceptualSimilarity/pull/139/files
@CugAgo Please provide an error location and if it's the same root cause, I'll submit a PRQ (if this is a 👍 issue for a first time contributor).
也许与此问题有关？https://github.com/richzhang/PerceptualSimilarity/pull/139/files
请提供错误位置，如果是相同的根本原因，我将提交 PRQ（如果这是👍第一次贡献者的问题）。
yes, I'm sorry for forgetting to provide accurate error information yesterday. In the normalize_tensor function, I think this bug can be fixed through this method, good job!
May I ask when this source code will be updated to the MONAI library? Can I update the corresponding version through pip