Poisson loss function

I think that Poisson loss function is not supported by default in PyTorch. You may try this:

BTW, I think that one should focus on loss functions that can manage learning rates self adaption instead of probabilities.