torch.utils.checkpoint

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

API名称

是否支持

限制与说明

torch.utils.checkpoint.checkpoint

否

   

torch.utils.checkpoint.checkpoint_sequential

      
父主题： PyTorch2.0