Pascal run time error

Getting:

in _unpad(self, bbox_tgt, clas_tgt)
19
20 def _unpad(self, bbox_tgt, clas_tgt):
—> 21 i = torch.min(torch.nonzero(clas_tgt-self.pad_idx))
22 return tlbr2cthw(bbox_tgt[i:]), clas_tgt[i:]-1+self.pad_idx
23

RuntimeError: operation does not have an identity.

When running learn.lr_find() or learn.fit_one_cycle(5, 1e-4)

The thread below might help. Good luck.