SemSeg with Pytorch