Ibrahim Bashir

icon indicating a website link www.linkedin.com/in/ibrahimxbashir

icon location Jerusalem icon location HUJI MSc Student in Statistics, background also in Data Science and Philosophy

Results 1 comments of Ibrahim Bashir

running under python in cpu only mode

You need to specify the device when loading the model and when using predict(). So set device='cpu' into both and it should work.