Adds grad-cam support on classifiers

conda/meta.yaml

@@ -24,18 +24,21 @@ requirements:
     - python >=3.10
     - pip
     - click {{ click }}
+    - grad-cam {{ grad_cam }}
     - matplotlib {{ matplotlib }}
     - numpy {{ numpy }}
     - pillow {{ pillow }}
     - psutil {{ psutil }}
     - pytorch {{ pytorch }}
+    - scikit-image {{ scikit_image }}
     - scikit-learn {{ scikit_learn }}
     - scipy {{ scipy }}
     - tabulate {{ tabulate }}
     - torchvision  {{ torchvision }}
     - tqdm {{ tqdm }}
     - tensorboard {{ tensorboard }}
-    - lightning >=2.0.3
+    - lightning {{ lightning }}
+    - lightning >=2.1.0,!=2.1.3
     - clapper
   run:
     - python >=3.10
@@ -45,13 +48,15 @@ requirements:
     - {{ pin_compatible('pillow') }}
     - {{ pin_compatible('psutil') }}
     - {{ pin_compatible('pytorch') }}
+    - {{ pin_compatible('scikit-image') }}
     - {{ pin_compatible('scikit-learn') }}
     - {{ pin_compatible('scipy') }}
     - {{ pin_compatible('tabulate') }}
     - {{ pin_compatible('torchvision') }}
     - {{ pin_compatible('tqdm') }}
     - {{ pin_compatible('tensorboard') }}
-    - {{ pin_compatible('lightning') }}
+    - {{ pin_compatible('lightning', max_pin='x.x') }}
+    - lightning >=2.1.0,!=2.1.3
     - clapper
 
 test: