models/compressive.py [216:224]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def build_layers(self):
        self.layers = nn.ModuleList()
        for l in range(self.args.nlayers):
            self.layers.append(TransformerSeqLayer(self.args, l))

    def get_layer(self, layer_ind):
        return self.layers[layer_ind]

    def forward(self, x, h_prev, target=None):
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



models/transformer_seq.py [214:222]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def build_layers(self):
        self.layers = nn.ModuleList()
        for l in range(self.args.nlayers):
            self.layers.append(TransformerSeqLayer(self.args, l))

    def get_layer(self, layer_ind):
        return self.layers[layer_ind]

    def forward(self, x, h_prev, target=None):
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -