Skip to content

如何加入重计算机制(torch.utils.checkpoint) #57

@ReyonRen

Description

@ReyonRen

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions