fix: gpu issue

explain.py

@@ -132,7 +132,7 @@ def save_explanation_results(res, path):
 
 
 model, image_transform = get_model(model_name)
-model.load_state_dict(torch.load(model_path))
+model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
 model.train()
 target_layers = [model.conv_head]
 gbp_model = GuidedBackpropReLUModel(model=model, device="cpu")