pytorch_pfn_extras.utils.checkpoint.checkpoint#

pytorch_pfn_extras.utils.checkpoint.checkpoint(function, *args, **kwargs)#

Parameters:

function (Module) –
args (Any) –
kwargs (Any) –

Return type:

Any