test gpu to cpu(.cpu()）is very slow
GuoxingYan opened this issue · 0 comments
GuoxingYan commented
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 opened this issue · 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