Thank you for your reply. Yeah, when I wanted to change the individual layers, I get a different output than when I’m running the whole network together. The segmentation.py file has a list of boolean variables which are used to change strided convolutions with dilated ones, which I don’t know how to do it while accessing individual modules of the segmentation network. Also, I just want to add a layer to the end. I guess I have to write the whole network again.