microsoft / tensorwatch

Debugging, monitoring and visualization for Python Machine Learning and Data Science
MIT License
3.41k stars 362 forks source link

Added arg device in get_saliency and get_image_saliency_results #77

Closed kikusui6192 closed 1 year ago

kikusui6192 commented 3 years ago

You can have device option when use get_saliency and get_image_saliency_results if have more than 2 GPUs in dev env.