a968bc8ae127d1e3833c2fa468858f9c48f6923c,batchflow/models/torch/encoder_decoder.py,EncoderModule,forward,#EncoderModule#Any#,22

Before Change

        self._make_modules(inputs, **kwargs)

    def forward(self, x):
        b_counter, d_counter = 0, 0
        outputs = []

        for _ in range(self.num_stages):
            for letter in self.encoder_layout:
                if letter in ["b"]:
                    x = self.encoder_b[b_counter](x)
                    b_counter += 1
                elif letter in ["d", "p"]:
                    x = self.encoder_d[d_counter](x)
                    d_counter += 1
                elif letter in ["s"]:
                    outputs.append(x)
        outputs.append(x)

        if self.return_all:
            return outputs

After Change

    def forward(self, x):
        outputs = []

        for letter, layer in zip(self.layout, self.values()):
            if letter in ["b", "d", "p"]:
                x = layer(x)
            elif letter in ["s"]:
                outputs.append(x)
        outputs.append(x)

        if self.return_all:
            return outputs

Instances