Closed daviddavid closed 2 years ago
Additional python requirements that are needed, but not listed, are: cython and torchvision
cython
torchvision
Also, and more importantly, this project requires a cuda compatible GPU to run.
Just added the mentioned libraries. You can comment out .cuda() and load the pretrained weights into CPU, if you don't have a GPU device.
Additional python requirements that are needed, but not listed, are:
cython
andtorchvision
Also, and more importantly, this project requires a cuda compatible GPU to run.