Image Segmentation Pytorch Cuda