Checkpoint with no grad requiring inputs PROBLEM

autograd
16

May I ask if there is any convenient solution for torch.utils.checkpoint.checkpoint_sequential? Thanks!