a7bbb96360a27fdc003952dd0ebf4c0db6328cfd,batchflow/models/torch/encoder_decoder.py,Decoder,body,#Any#Any#,270

Before Change

    def body(cls, inputs, **kwargs):
        kwargs = cls.get_defaults("body", kwargs)
        decoder = kwargs.pop("decoder")
        return DecoderModule(inputs=inputs, **{**kwargs, **decoder})

    @classmethod
    def head(cls, inputs, target_shape, classes, **kwargs):
        kwargs = cls.get_defaults("head", kwargs)

After Change

    def body(cls, inputs, **kwargs):
        kwargs = cls.get_defaults("body", kwargs)
        decoder = kwargs.pop("decoder")
        layers = [("decoder", DecoderModule(inputs=inputs, **{**kwargs, **decoder}))]
        return nn.Sequential(OrderedDict(layers))

    @classmethod
    def head(cls, inputs, target_shape, classes, **kwargs):
        kwargs = cls.get_defaults("head", kwargs)

Instances