b31cab02dcf9f2c75974ade863b0b036387a6e56,batchflow/models/torch/encoder_decoder.py,EncoderBlock,__init__,#EncoderBlock#Any#Any#Any#Any#Any#,190

Before Change

class EncoderBlock(nn.Module):
    def __init__(self, inputs, i, steps, downsample, block_args, **kwargs):
        super().__init__()
        ifilters = list(get_shape(inputs))[1]
        self.downsample = ConvBlock(inputs, filters=ifilters, **{**kwargs, **downsample})
        shape = list(get_shape(self.downsample))
        shape = tuple(shape)

        base_block = block_args.get("base")
        args = {**kwargs, **block_args, **unpack_args(block_args, i, steps)}
        self.encoder = base_block(shape, **args)

After Change

        self.encoder = base_block(inputs, **args)

        e_shape = get_shape(self.encoder)
        ifilters = downsample.get("filters") or e_shape[1]
        self.downsample = ConvBlock(e_shape, filters=ifilters, **{**kwargs, **downsample})

        self.output_shape = self.downsample.output_shape

Instances