search

sacmehta / EdgeNets

This repository contains the source code of our work on designing efficient CNNs for computer vision
MIT License
411 stars 82 forks source link

Runtime Error #2

Closed Ideal-Bai closed 5 years ago

Ideal-Bai commented 5 years ago

Sorry for disturbing you. When I run the test_detection.py ,I got an error:RuntimeError: expected backend CUDA and dtype Float but got backend CPU and dtype Float. How can I solve it? Thank you

sacmehta commented 5 years ago

Let me check it. Meanwhile, you can use CUDA_VISIBLE_DEVICES=‘’. This will enforce your model to run on CPU.

sacmehta commented 5 years ago

Are you using torchvision version 0.3? If not, could you try with that?

sacmehta commented 5 years ago

I have fixed it. Please check