peeyushsinghal commited on
Commit
029ab07
·
1 Parent(s): 29aa270

set cuda = False for grad cam

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -167,7 +167,7 @@ def process_gradcam_images(num_images,layer,opacity,image_list=None):
167
  selected_data = [image_list]
168
 
169
  layer_model = dict_layer.get(layer)
170
- cam = GradCAM(model=loaded_model, target_layers = [layer_model], use_cuda = True)
171
  grad_images = []
172
  inv_normalize = transforms.Normalize(
173
  mean=[-0.50/0.2197, -0.50/0.1858, -0.50/0.1569], # mean_ds = [0.2197, 0.1858, 0.1569]
 
167
  selected_data = [image_list]
168
 
169
  layer_model = dict_layer.get(layer)
170
+ cam = GradCAM(model=loaded_model, target_layers = [layer_model], use_cuda = False)
171
  grad_images = []
172
  inv_normalize = transforms.Normalize(
173
  mean=[-0.50/0.2197, -0.50/0.1858, -0.50/0.1569], # mean_ds = [0.2197, 0.1858, 0.1569]