| 936 | torch.set_grad_enabled(False) |
| 937 | |
| 938 | class LeakyReLU0(Module): |
| 939 | def __init__(self): |
| 940 | super().__init__() |
| 941 | self.leakyrelu = torch.nn.LeakyReLU(0.02) |
| 942 | |
| 943 | def forward(self, input): |
| 944 | return self.leakyrelu(input) |
| 945 | |
| 946 | class LeakyReLU1(Module): |
| 947 | def forward(self, input): |
no outgoing calls
searching dependent graphs…