Skip to content

AD1024/torch-checkpointing

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Torch Checkpointing

Convert a torch model to a checkpointed model.

Reference

Tianqi Chen, Bing Xu, Chiyuan Zhang and Carlos Guestrin, "Training Deep Nets with Sublinear Memory Cost". Apr. 22, 2016. https://arxiv.org/abs/1604.06174.

HiddenLayer

About

Compile a torch model to a checkpointed model

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages