pytorch_segmentation only for single gpu and cpu usage I think this code easy to be used,although I do not write the usage in the readme.