Searched refs:NoGradModule (Results 1 – 3 of 3) sorted by relevance
597 class NoGradModule(torch.nn.Module): class609 s = torch.jit.script(NoGradModule())
1382 class NoGradModule(nn.Module): class1396 NoGradModule().float().to(device_id),
7144 class NoGradModule(torch.nn.Module): class7154 module = NoGradModule()