a968bc8ae127d1e3833c2fa468858f9c48f6923c,batchflow/models/torch/encoder_decoder.py,DecoderModule,_make_modules,#DecoderModule#Any#,116

Before Change

        upsample_args = kwargs.pop("upsample")
        combine_args = kwargs.pop("combine")

        self.decoder_b, self.decoder_u, self.decoder_c = nn.ModuleList(), nn.ModuleList(), nn.ModuleList()

        for i in range(num_stages):
            for letter in decoder_layout:
                if letter in ["b"]:

After Change

                else:
                    raise ValueError("Unknown letter in order {}, use one of ("b", "u", "c")".format(letter))

                self.update([(layer_desc, layer)])
                self.layout += letter

Instances