Giters

nabsabraham / focal-tversky-unet

This repo contains the code for our paper "A novel focal Tversky loss function and improved Attention U-Net for lesion segmentation" accepted at IEEE ISBI 2019.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

loss: Nan.

linyingyingkarina opened this issue · comments

commented

Hi, thanks for your work.

I have succeeded to run the code. But when I ran it in a new GPU, the data, the code, all the same. However, during the training, even in the first epoch. The loss is shown with Nan. Do you know why? or any possible of causing this??

Thanks a lot.

commented

Having the same issue. Any success lately?

commented

Same issue. This can be avoided when gamma is set to 1.