Merge branch 'f/experiment_logging' into 'main'

Add support for pytorch finetuning callback on image models See merge request es/ai/hannah/hannah!376
ekut-es · Apr 8, 2024 · 6b6dc27 · 6b6dc27
2 parents 4a7d742 + a57e8c2
commit 6b6dc27
Show file tree

Hide file tree

Showing 2 changed files with 248 additions and 246 deletions.
diff --git a/hannah/modules/base.py b/hannah/modules/base.py
@@ -85,6 +85,7 @@ def __init__(
     ) -> None:
         super().__init__()
 
+        self.model = None
         self.model = None
         ignore = None
         if isinstance(model, nn.Module):