Skip to content

Commit

Permalink
edit tab to 4 spaces or will get TabError
Browse files Browse the repository at this point in the history
  • Loading branch information
flyingpot authored Jan 5, 2018
1 parent aacfb89 commit f40b2c7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion grad-cam.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def __call__(self, x):
for name, module in self.model._modules.items():
x = module(x)
if name in self.target_layers:
x.register_hook(self.save_gradient)
x.register_hook(self.save_gradient)
outputs += [x]
return outputs, x

Expand Down

0 comments on commit f40b2c7

Please sign in to comment.