search

zzzqzhou / RAM-DSIR

[ECCV'22] Generalizable Medical Image Segmentation via Random Amplitude Mixup and Domain-Specific Image Restoration
37 stars 3 forks source link

nan or inf found in input tensor #4

Closed minkkkey closed 1 year ago

minkkkey commented 1 year ago

This problem occurred while training the prostate dataset. How to solve it? Looking forward your reply. Thanks in advance. i use pytorch==1.8

zzzqzhou commented 1 year ago

Hi, this problem may caused by KD consistency loss. You can use MSE consistency loss when train on the prostate dataset. This will not make much difference to the result.

minkkkey commented 1 year ago

Hi, this problem may caused by KD consistency loss. You can use MSE consistency loss when train on the prostate dataset. This will not make much difference to the result.

Thank you so much!!!!!!!!!!!!!!!