twentyfiveYang

Hello Abraham, I'm also trying to do the same thing and find a way to train our own images. The dataset structure required here should be like:  The way...

Hi @universewill ,I have the same problem as yours. Have you find the solution?

> Do you see nan loss during the training? Based on what I've tested, if you turn off amp=True, it resolves nan loss under certain cases (with the cost of...

Encounter the same problem. Did you find the solution for this? I also tried to visualize the grad-cam map of the Unet++ model.