torch.utils.checkpoint 若API未标明支持情况则代表该API暂无验证结果,待验证后更新表格。 API名称 是否支持 限制与说明 torch.utils.checkpoint.checkpoint 是 torch.utils.checkpoint.checkpoint_sequential 是 支持fp32 父主题: PyTorch原生API