Method __init__

src/shape_assembly/models/train/transformer.py:49–52

(self, layer, N)

class Decoder(pl.LightningModule):
    def __init__(self, layer, N):
        super().__init__()
        self.layers = clones(layer, N)
        self.norm = LayerNorm(layer.size)

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