torch.utils.checkpoint
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.utils.checkpoint.checkpoint |
是 |
|
torch.utils.checkpoint.checkpoint_sequential |
是 |
支持fp32 |
父主题： PyTorch原生API
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.utils.checkpoint.checkpoint |
是 |
|
torch.utils.checkpoint.checkpoint_sequential |
是 |
支持fp32 |