Skip to content

Commit

Permalink
Add changes to work better with pytorch-grad-cam
Browse files Browse the repository at this point in the history 
Changes classifier base class to extend nn.Module and implements the
forward() method so the classifiers can be used with pytorch-grad-cam.

Fixes #45
  • Loading branch information
@johnbradley
johnbradley committed Nov 20, 2024
1 parent 12c503d commit 952e5a3
Show file tree
Hide file tree
Showing 3 changed files with 327 additions and 3 deletions.
Loading

0 comments on commit 952e5a3

Please sign in to comment.