Searched refs:ModelWithCheckpointSubmodule (Results 1 – 1 of 1) sorted by relevance
300 class ModelWithCheckpointSubmodule(nn.Module): class318 self.checkpoint1 = ModelWithCheckpointSubmodule(checkpoint, use_reentrant)319 self.checkpoint2 = ModelWithCheckpointSubmodule(checkpoint, use_reentrant)