fix: Fix deepcopy removing model signature wrapper

CHANGELOG.md

@@ -2,6 +2,12 @@
 # Changelog
 All notable changes to this project will be documented in this file.
 
+### [2.0.2]
+
+#### Fixed
+
+- Fix deepcopy removing model signature wrapper from the model for classification
+
 ### [2.0.1]
 
 #### Fixed

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quadra"
-version = "2.0.1"
+version = "2.0.2"
 description = "Deep Learning experiment orchestration library"
 authors = [
   "Federico Belotti <federico.belotti@orobix.com>",

quadra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.0.1"
+__version__ = "2.0.2"
 
 
 def get_version():

quadra/tasks/classification.py

@@ -338,7 +338,8 @@ def generate_report(self) -> None:
             self.datamodule.setup(stage="test")
 
         # Deepcopy to remove the inference mode from gradients causing issues when loading checkpoints
-        self.module.model = deepcopy(self.module.model)
+        # TODO: Why deepcopy of module model removes ModelSignatureWrapper?
+        self.module.model.instance = deepcopy(self.module.model.instance)
         if "16" in self.trainer.precision:
             log.warning("Gradcam is currently not supported with half precision, it will be disabled")
             self.module.gradcam = False