search

CSAILVision / semantic-segmentation-pytorch

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
http://sceneparsing.csail.mit.edu/
BSD 3-Clause "New" or "Revised" License
4.95k stars 1.1k forks source link

test gpu to cpu(.cpu()）is very slow #255

Open GuoxingYan opened 3 years ago

GuoxingYan commented 3 years ago

https://github.com/CSAILVision/semantic-segmentation-pytorch/blob/master/test.py 82 line pred = as_numpy(pred.squeeze(0).cpu()) .cpu()is very slow