quadra 2.2.2__py3-none-any.whl → 2.2.3__py3-none-any.whl

quadra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.2.2"
+__version__ = "2.2.3"
 
 
 def get_version():

quadra/modules/classification/base.py

@@ -138,13 +138,9 @@ class ClassificationModule(BaseLightningModule):
         if isinstance(self.model.features_extractor, timm.models.resnet.ResNet):
             target_layers = [cast(BaseNetworkBuilder, self.model).features_extractor.layer4[-1]]
 
-            # Get model current device
-            device = next(self.model.parameters()).device
-
             self.cam = GradCAM(
                 model=self.model,
                 target_layers=target_layers,
-                use_cuda=device.type == "cuda",
             )
             # Activating gradients
             for p in self.model.features_extractor.layer4[-1].parameters():
@@ -262,7 +258,7 @@ class MultilabelClassificationModule(BaseLightningModule):
                 self.gradcam = False
             else:
                 target_layers = [model[0].features_extractor.layer4[-1]]
-                self.cam = GradCAM(model=model, target_layers=target_layers, use_cuda=torch.cuda.is_available())
+                self.cam = GradCAM(model=model, target_layers=target_layers)
 
     def forward(self, x):
         return self.model(x)

quadra/tasks/classification.py

@@ -1138,7 +1138,6 @@ class ClassificationEvaluation(Evaluation[ClassificationDataModuleT]):
             self.cam = GradCAM(
                 model=self.deployment_model.model,
                 target_layers=target_layers,
-                use_cuda=(self.device != "cpu"),
             )
             for p in self.deployment_model.model.features_extractor.layer4[-1].parameters():
                 p.requires_grad = True

quadra/utils/models.py

@@ -131,7 +131,6 @@ def get_feature(
             cam = GradCAM(
                 model=feature_extractor,
                 target_layers=target_layers,
-                use_cuda=torch.cuda.is_available(),
             )
             for p in feature_extractor.features_extractor.layer4[-1].parameters():
                 p.requires_grad = True

{quadra-2.2.2.dist-info → quadra-2.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quadra
-Version: 2.2.2
+Version: 2.2.3
 Summary: Deep Learning experiment orchestration library
 Home-page: https://orobix.github.io/quadra
 License: Apache-2.0
@@ -21,9 +21,9 @@ Classifier: Topic :: Software Development :: Libraries
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Provides-Extra: onnx
 Requires-Dist: albumentations (>=1.3,<1.4)
-Requires-Dist: anomalib-orobix (==0.7.0.dev140)
+Requires-Dist: anomalib-orobix (==0.7.0.dev141)
 Requires-Dist: boto3 (>=1.26,<1.27)
-Requires-Dist: grad_cam (==1.4.6)
+Requires-Dist: grad-cam-orobix (==1.5.3.dev001)
 Requires-Dist: h5py (>=3.8,<3.9)
 Requires-Dist: hydra_colorlog (>=1.2,<1.3)
 Requires-Dist: hydra_core (>=1.3,<1.4)

{quadra-2.2.2.dist-info → quadra-2.2.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-quadra/__init__.py,sha256=3IxXq2HWhVPGbI13XyFjCXt9xDBgUMxWYixTBEdYtOk,112
+quadra/__init__.py,sha256=k4RvIQHL_Vg6uYb-uXD4XswG0lCw26LdATU9LN4dpu8,112
 quadra/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/callbacks/anomalib.py,sha256=Z0Xx3M9UEMYGY0BpSsqemgQSQ6jAr4SrvlbDg_oG9C8,11913
 quadra/callbacks/lightning.py,sha256=1OTM6fB7qBVLPWNBAZJOb3B00q0kAxMWkPjTEn9YgF0,20182
@@ -230,7 +230,7 @@ quadra/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/modules/backbone.py,sha256=xiZBqgzr1S45GX9mydl29TFuahLDaHrU7Fy73LGIyGI,909
 quadra/modules/base.py,sha256=fS_kbOAfEihHFfiTcyuyDXTzkmkJr2ivtAMiZqjOUog,10303
 quadra/modules/classification/__init__.py,sha256=6keltBhC1yzgbNttBuykNYJAUMyOrY-HDNgGZGfI93I,141
-quadra/modules/classification/base.py,sha256=kC2vPHMEoeHyK_tECau9W9ynLMJzkI9cpOjs-DZeU1k,12119
+quadra/modules/classification/base.py,sha256=QdHtHY2tF_qh2wU01Oo0TWjh9CTqa46tyF4VgcLd__M,11937
 quadra/modules/ssl/__init__.py,sha256=oeUoGHrsESZ0595-JxPxURBP124jtNfrITbVovBpANA,302
 quadra/modules/ssl/barlowtwins.py,sha256=iW6f7ADSEkbs7z-88x680204-Ez-iF1Yd2SdQzcLpRY,1884
 quadra/modules/ssl/byol.py,sha256=8sviU7MS9MXDbS9Ogu9-qJCZrkTI2pPxgEwNvc7EqIo,7084
@@ -250,7 +250,7 @@ quadra/schedulers/warmup.py,sha256=chzzrK7OqqlicBCxiF4CqMYNrWu6nflIbRE-C86Jrw0,4
 quadra/tasks/__init__.py,sha256=tmAfMoH0k3UC7r2pNrgbBa1Pfc3tpLl3IObFF6Z0eRE,820
 quadra/tasks/anomaly.py,sha256=rl3F39kABt5deSLOeY2s_1t2mR4x1-42VdP0flRrZMs,24583
 quadra/tasks/base.py,sha256=5Rsjdothqb4YXMIN_s98HdoPmJBfQKV0ZwnyvUuihYI,14101
-quadra/tasks/classification.py,sha256=w3bq_EPgz3X---iaGBlVD9hb3ncWPClbrjm6496Heh8,52784
+quadra/tasks/classification.py,sha256=I4NEsNVn1lOu-HLaD_hqowfnedhD4l-wZ4JPI1eD4pg,52735
 quadra/tasks/patch.py,sha256=EJvbtvlebyOtk6m47juW8XMXr1v_bLFTIKqI1KC0HRA,20244
 quadra/tasks/segmentation.py,sha256=5GF7CZjm1dQGQ-Q0kAc68GSsNv6FTxBsl_rbEZQwjmU,16213
 quadra/tasks/ssl.py,sha256=SVZeAW5xVkfeGTwMqRhGKLagMC2kHrxoOaNEDzbCZ0A,20552
@@ -267,7 +267,7 @@ quadra/utils/imaging.py,sha256=Cz7sGb_axEmnGcwQJP2djFZpIpGCPFIBGT8NWVV-OOE,866
 quadra/utils/logger.py,sha256=tQJ4xpTAFKx1g-UUm5K1x7zgoP6qoXpcUHQyu0rOr1w,556
 quadra/utils/mlflow.py,sha256=7E09JShGm2qO7bLb_8srA_RYdVAudxeDYmX7pMdjoVU,3524
 quadra/utils/model_manager.py,sha256=P5JtY95p6giQ6mb4TUnWsNwUh5ClzHBillnG5SA56QY,12546
-quadra/utils/models.py,sha256=lJRnZQuxW8KRPdokIl3snBQX4e6IA2Yqil8-ZsIkXlA,19808
+quadra/utils/models.py,sha256=xLOT6Sorpc54dYn9XG8RpqoLupvNrMCiE_QvvIMLwHA,19756
 quadra/utils/patch/__init__.py,sha256=YenDdsI937kyAJiE0dP3_Xua8gHIoFjheoWMnpx_TGU,509
 quadra/utils/patch/dataset.py,sha256=hqM7XyPNDmI9_uJSrAxNYOasdNlEHcg-npP1S9Bb05Y,61374
 quadra/utils/patch/metrics.py,sha256=E1PeHFp10pPgkb6484fDvRLn2E9NDy9xIqEmMBeusOw,17644
@@ -293,8 +293,8 @@ quadra/utils/validator.py,sha256=eFCGr0ss1gYSpsL31JbsCXPZUMJAI9_H-mGodt6UGsU,466
 quadra/utils/visualization.py,sha256=UvGHX0dumfjpT_KX3Yc1W2B5sAXXaZZWIwXQAi4sdoQ,15950
 quadra/utils/vit_explainability.py,sha256=hY0awehj6UkyBhnBlW5uWoJTsBfgow5Nll9fAqrzmMo,13337
 hydra_plugins/quadra_searchpath_plugin.py,sha256=AAn4TzR87zUK7nwSsK-KoqALiPtfQ8FvX3fgZPTGIJ0,1189
-quadra-2.2.2.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
-quadra-2.2.2.dist-info/METADATA,sha256=zFQ9_EqPmEXtfYjuH6TLFCfu1zkbOV0v6RN5posMqng,17609
-quadra-2.2.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-quadra-2.2.2.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
-quadra-2.2.2.dist-info/RECORD,,
+quadra-2.2.3.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
+quadra-2.2.3.dist-info/METADATA,sha256=cRx79iZ9AKEE8bWu8gvMy_NSdp5KuM2IrZB_eGdQU5E,17623
+quadra-2.2.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+quadra-2.2.3.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
+quadra-2.2.3.dist-info/RECORD,,