| 23 | |
| 24 | |
| 25 | class _MLP(torch.nn.Module): |
| 26 | def __init__(self): |
| 27 | super().__init__() |
| 28 | self.l1 = torch.nn.Linear(8, 16) |
| 29 | self.l2 = torch.nn.Linear(16, 8) |
| 30 | self.l3 = torch.nn.Linear(8, 4) |
| 31 | |
| 32 | def forward(self, x): |
| 33 | return self.l3(self.l2(self.l1(x).relu()).relu()) |
| 34 | |
| 35 | |
| 36 | def _export(): |