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