albert_pytorch icon indicating copy to clipboard operation
albert_pytorch copied to clipboard

Published 20 hours ago • verified publisher icon lonePatient

如何加入重计算机制（torch.utils.checkpoint)

Open ReyonRen opened this issue 4 years ago • 0 comments

您好，感谢您的分享。 请问如果我想在代码中加入pytorch的重计算机制（torch.utils.checkpoint)来减少显存开销，应该怎么做呢？

ReyonRen avatar Sep 16 '20 11:09 ReyonRen