Adds grad-cam support on classifiers

conda/meta.yaml

@@ -38,7 +38,7 @@ requirements:
     - torchvision  {{ torchvision }}
     - tqdm {{ tqdm }}
     - tensorboard {{ tensorboard }}
-    - lightning >=2.0.3
+    - lightning {{ lightning }}
     - clapper
   run:
     - python >=3.10
@@ -54,7 +54,7 @@ requirements:
     - {{ pin_compatible('torchvision') }}
     - {{ pin_compatible('tqdm') }}
     - {{ pin_compatible('tensorboard') }}
-    - {{ pin_compatible('lightning') }}
+    - {{ pin_compatible('lightning', max_pin='x.x') }}
     - clapper
 
 test: