torch.utils.checkpoint

若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。

API名称

是否支持

限制与说明

torch.utils.checkpoint.checkpoint

是

-

torch.utils.checkpoint.checkpoint_sequential

是

支持fp32
父主题： PyTorch原生API支持度