(self, x)
| 288 | self.norm = nn.LayerNorm(output_dim) |
| 289 | |
| 290 | def forward(self, x): |
| 291 | x = F.gelu(self.layer1(x)) |
| 292 | x = F.gelu(self.layer2(x)) |
| 293 | x = F.gelu(self.layer3(x)) |
| 294 | x = F.gelu(self.layer4(x)) |
| 295 | x = self.layer5(x) |
| 296 | x = self.norm(x) |
| 297 | return x |
| 298 | |
| 299 | class OneShotDataset(Dataset): |
| 300 | def __init__(self, original_dataset): |
nothing calls this directly
no outgoing calls
no test coverage detected