Is there a way to extract the segmentation model only?

[AwayFromKeyboardwarrior]

I have my own way to call the dataloader and sampler, loss, etc. is there a way to extract the segmentation model only, so that I could plug it into my framework and run it? It is connected to so many other functions and called from them ;( Can someone please help me?