self-supervised-depth-completion icon indicating copy to clipboard operation
self-supervised-depth-completion copied to clipboard

Fixing a bug of mixing data stored on gpu and a model stored on cpu w…

Open mac137 opened this issue 4 years ago • 0 comments

my PR fixes the following bug: RuntimeError: expected device cuda:0 but got device cpu

... while running this command: python main.py --cpu --evaluate [path2checkpoint]

mac137 avatar May 15 '20 14:05 mac137