Pelee.Pytorch icon indicating copy to clipboard operation
Pelee.Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon yxlijun

Disabling GPU

Open kunalgoyal9 opened this issue 5 years ago • 1 comments

When I disable GPU in configs/Pelee_VOC.py test_cfg.cuda = false. I got RuntimeError: expected device cpu but got device cuda:0.

So some model is using GPU. But I am not able to figure which one.

kunalgoyal9 avatar Nov 19 '19 01:11 kunalgoyal9

Meet the same problem. Have you solved it?

Z-Z-Han avatar Oct 31 '20 02:10 Z-Z-Han