| 889 | torch.set_grad_enabled(False) |
| 890 | |
| 891 | class Softplus0(torch.nn.Module): |
| 892 | def __init__(self): |
| 893 | super().__init__() |
| 894 | self.softplus = torch.nn.Softplus(1.0, 20.0) |
| 895 | |
| 896 | def forward(self, x): |
| 897 | return self.softplus(x) |
| 898 | |
| 899 | class Softplus1(Module): |
| 900 | def forward(self, input): |
no outgoing calls
searching dependent graphs…