torch.utils.checkpoint

若API未标明支持情况,则代表该API的支持情况待验证。

API名称

是否支持

限制与说明

torch.utils.checkpoint.checkpoint

  

torch.utils.checkpoint.checkpoint_sequential

支持fp32