How to train instance segmented custom data on Pytorch

[Bombardelli]

I am trying to train an instance segmented data on PyTorch. I created the dataset on Roboflow and I am having difficulties creating the training loop for a segmented data. I have watched the training loop lessons for classification but segmented data is looking more complex. I already transformed and augmented the data based on some tutorials on the PyTorch website.