ensure the model is running on CPU, resolve some rerun issue

rigvedrs / YOLO-V8-CAM

Wanna know what your model sees? Here's a package for applying EigenCAM on the new YOLO V8 model

MIT License

131 stars 26 forks source link

ensure the model is running on CPU, resolve some rerun issue #23

Closed kimyd closed 1 month ago

kimyd commented 1 month ago

The EigenCAM is computed on the CPU, so to ensure model is also on the CPU, added model.cpu() This should resolve 'rerun' issues as issues #14 for me.

rigvedrs commented 1 month ago

Thanks for pointing that out. Have added it