Searched refs:SaveForBwdModule (Results 1 – 1 of 1) sorted by relevance
185 class SaveForBwdModule(torch.nn.Module): class325 model = SaveForBwdModule()