Hi @ptrblck
When I try to use the loss function in a train loop, the results come out as nan
loss = loss_fn(output_batch, labels_batch)
logger.info("loss shape: {}, loss data: {}".format(loss.shape, loss.data.item()))

train_unet:train:87: loss shape: torch.Size([]), loss data: nan
I unit tested the implementation using some test matrices for predictions and labels, the implementation is okay.
Any idea what could cause this? It should be some large value at the start of training.