Open lusihua opened 4 years ago
RuntimeError Traceback (most recent call last)
我也遇到这样的问题在test_on_images.ipynb
hello, did you solve it?
I solved it by passing in the device as an argument for the run_first_stage function and adding .to(device) the img variable before feeding it into the network. I think their original code is used only for running on the CPU. To run the inference with GPU, you should modify this part of the run_first_stage function.
run get_aligned_face_from_mtcnn.ipynb error occurred