zhangabner

same problem here. 16 vis = GradCam(module, device) 17 ---> 18 model_outs[name] = list(map(lambda x: tensor2img(vis(x, None,postprocessing=image_net_postprocessing)[0]), inputs)) 19 del module 20 torch.cuda.empty_cache() ~\Documents\A-journey-into-Convolutional-Neural-Network-visualization-\visualisation\core\GradCam.py in __call__(self, input_image, layer, guide, target_class,...