Method forward

src/shape_assembly/models/train/transformer.py:54–57

(self, x, memory, src_mask, tgt_mask)

        self.norm = LayerNorm(layer.size)

    def forward(self, x, memory, src_mask, tgt_mask):
        for layer in self.layers:
            x = layer(x, memory, src_mask, tgt_mask)
        return self.norm(x)

class LayerNorm(pl.LightningModule):
    def __init__(self, features, eps=1e-6):