Searched defs:checkpoint_wrapper (Results 1 – 4 of 4) sorted by relevance
197 def checkpoint_wrapper( function
20 import torch.distributed.algorithms._checkpoint.checkpoint_wrapper as checkpoint_wrapper namespace
21 def checkpoint_wrapper(fn): function
39 def checkpoint_wrapper(fn): function