loss diverges to NaN

[minxueric]

In the _create_loss function, the loss_breslow is something divided by Obs. However the variable Obs is the count of events in a batch, which is likely to be 0.

How do you solve this problem?

[liupei101]

Maybe you can try to add some codes to manually skip the batch if you found that there is no event observation in this batch.