4c50261df4847bdfd7c8067307e8532f96d04104,batchflow/models/torch/encoder_decoder.py,EncoderDecoder,decoder,#Any#Any#,147

Before Change

        steps = kwargs.pop("num_stages", len(inputs)-1)
        factor = kwargs.pop("factor")

        if isinstance(factor, int):
            factor = int(factor ** (1/steps))
            factor = [factor] * steps
        elif not isinstance(factor, list):
            raise TypeError("factor should be int or list of int, but %s was given" % type(factor))

        x = inputs[-1]
        for i in range(steps):
            x = cls.upsample(x, factor=factor[i], **kwargs)

After Change

        x = inputs[-1]
        decoders = []
        for i in range(steps):
            x = DecoderBlock(x, inputs[-i-3], i, upsample, block_args, **kwargs)
            decoders.append(x)
        return decoders

Instances