Instance Segmentation Pytorch Lightning