Closed timmermansjoy closed 1 year ago
I developed this project on GPU. I have not tested it on CPU. If you want to run the code on the specific GPU, please use CUDA_VISIBLE_DEVICES
.
Okay, I'm having trouble getting it up and running due to the fact all those datasets seem to be broken / not available for download
Please check README. I have updated the download link. You can download VL-CMU-CD-binary255
at here.
thanks!
in the main train.py script you can pass the device param. but there is a lot of stuff cuda Is used for and can't use the CPU to do anything? is this meant to do stuff like
cuda:4
or so?