Just use `torch.inf`, as torch._six is gone after https://github.com/pytorch/pytorch/pull/94709
Attach a file by drag & drop or click to upload
