(self, x)
| 83 | self.second_layers = nn.Sequential(*second_layer_list) |
| 84 | |
| 85 | def forward(self, x): |
| 86 | if self.posenc > 0: |
| 87 | x = self.PE(x) |
| 88 | h = self.first_layers(x) |
| 89 | h = torch.cat([x, h], dim=-1) |
| 90 | h = self.second_layers(h) |
| 91 | return h |
| 92 | |
| 93 | |
| 94 | class SiLU(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected