semantic-segmentation-pytorch icon indicating copy to clipboard operation
semantic-segmentation-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon CSAILVision

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

Open GuoxingYan opened this issue 4 years ago • 0 comments

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

GuoxingYan avatar Dec 14 '20 08:12 GuoxingYan