(self, x)
| 64 | self.eps = eps |
| 65 | |
| 66 | def forward(self, x): |
| 67 | mean = x.mean(-1, keepdim=True) |
| 68 | std = x.std(-1, keepdim=True) |
| 69 | return self.a_2 * (x - mean) / (std + self.eps) + self.b_2 |
| 70 | |
| 71 | class SublayerConnection(pl.LightningModule): |
| 72 | def __init__(self, size): |
nothing calls this directly
no outgoing calls
no test coverage detected