tensorwatch
tensorwatch copied to clipboard
Added arg device in get_saliency and get_image_saliency_results
You can have device option when use get_saliency and get_image_saliency_results if have more than 2 GPUs in dev env.