Pytorch Medical Image Segmentation