Method __init__

src/shape_assembly/models/train/transformer.py:38–41

(self, layer, N)

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

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