Searched refs:torch_utils_checkpoint (Results 1 – 2 of 2) sorted by relevance
12 from torch.utils.checkpoint import checkpoint as torch_utils_checkpoint unknown133 torch_utils_checkpoint,
123 torch_utils_checkpoint = torch.utils.checkpoint.checkpoint126 checkpoint_fn=torch_utils_checkpoint,