torch.utils.checkpoint

Note

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

API名称 是否支持 限制与说明
torch.utils.checkpoint.checkpoint -
torch.utils.checkpoint.checkpoint_sequential 支持fp32