Inference on CPU

[Hamdi-Ben-Abdallah]

Hello,

I would like to test your code and your pre-trainer model and I wanted to know if it works only on CPU?
Do you know the execution time to process an image?

Thanks in advance.

[CauchyComplete]

Hi,
I'm sorry to inform you that this code cannot run without CUDA. You'll see so many .cuda() in the whole project.
If you wish to run it, you need to manually modify the entire project.