Nan loss when training

[sydat2701]

Hi, thanks for your great job!
Currently, I got a problem when training. My TRE loss always got NAN and the training log is as shown in the figure. Do you have any idea for this? I downloaded the dataset and only resample to your required size in paper (and did not do any other preprocessing steps as I saw that original BraTS dataset was run skull-stripped).
Thank for your help!

[cwmok]

Hi @sydat2701,

I never encounter this issues. Is the resulting warped image looks good after training?