Searched refs:CheckpointOnceModule (Results 1 – 1 of 1) sorted by relevance
465 class CheckpointOnceModule(nn.Module): class in CommonDistributedDataParallelTest481 class CheckpointTwiceModule(CheckpointOnceModule):552 self.CheckpointOnceModule(use_reentrant=use_reentrant),561 self.CheckpointOnceModule(),587 self.CheckpointOnceModule(use_reentrant=use_reentrant),594 self.CheckpointOnceModule(use_reentrant=use_reentrant),