About training loss

[Ltohka]

I followed readme to reproduce the project and when I do stage 2 training, the loss contains nan.

if torch.isnan(depth_pred).any():
                    logging.warning("model_pred contains NaN.")

I put this code after this line, and its output shows that this is a nan.

So Where did I go wrong?