CUDA error: an illegal memory access was encountered

[18022443868]

Can't this loss be used for RGB as input for image training

[18022443868]

The above CUDA errors occur in training

[yatengLG]

Please provide the running environment and part of the loss function code.

another way, you can run this example to verify the loss function code. https://github.com/yatengLG/Focal-Loss-Pytorch/blob/master/Demo.ipynb