Custom loss function

[j-adamczyk]

As far far I understand, some methods, e.g. ROAR, theoretically can use any reasonable, differentiable loss function. BCE was used in the paper, but we could use any other reasonable function. It would be nice to be able to use any loss function from PyTorch instead of only "BCE" and "MSE" strings in CARLA.