def forward(self, x):
        """Sequentially pass `x` trough model`s encoder, decoder and heads"""
        features = self.encoder(x)
        decoder_output = self.decoder(*features)
        output = self.classifier(decoder_output)

        return output
Sequentially pass `x` trough model`s encoder, decoder and heads
code/networks/efficientunet.py