GPU for Examples_core.ipynb

PAIR-code / saliency

Framework-agnostic implementation for state-of-the-art saliency methods (XRAI, BlurIG, SmoothGrad, and more).

https://pair-code.github.io/saliency/

Apache License 2.0

950 stars 191 forks source link

Open betterze opened 1 year ago

betterze commented 1 year ago

Dear PAIR code group,

Thank you for sharing this great work. I really like it.

I am using the Examples_core.ipynb now. Is there a way to use GPU in this torch implementation? The cpu one is a bit slow.

Thank you for your help.

Best Wishes,

Zongze