Method forward

src/shape_assembly/models/train/transformer.py:43–46

(self, x, mask)

        self.norm = LayerNorm(layer.size)

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

class Decoder(pl.LightningModule):
    def __init__(self, layer, N):