returnn.torch.util.exception_helper

Helper for any type of PyTorch exceptions

returnn.torch.util.exception_helper.help_on_torch_exception(exc: Exception, *, step_idx: int | None = None, extern_data: TensorDict | None = None, model: Module | Module, always_direct_print: bool = False)

Gather some information which might be helpful for debugging a PyTorch exception.