PyTorch Forums
Does torch.utils.checkpoint support custom cuda operators?
dihe_huang
(dihe huang)
June 15, 2022, 8:04am
1
Does torch.utils.checkpoint support custom cuda operators? Thanks!