self-supervised-depth-completion
self-supervised-depth-completion copied to clipboard
Fixing a bug of mixing data stored on gpu and a model stored on cpu w…
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]