Hello, I am implementing a segmentation task with masks and images in input and one class in output. I am trying to use the binary cross entropy from pytorch but I have this error :
> RuntimeError: “binary_cross_entropy” not implemented for 'Long’
Indeed, I have formatted my mask as a long type, otherwise this message appear from the loss :
> RuntimeError: Found dtype Float but expected Long
Does someone already have this error and know from where it come from ?
Note that I am in torch ‘1.6.0’
Thank you in advance