diff --git a/sgmse/model.py b/sgmse/model.py
index 2baed0e1205cd473823b7863bad9101896e8192f..a97a56680b44a397827ad21afecfdf57a6ad4b0c 100644
--- a/sgmse/model.py
+++ b/sgmse/model.py
@@ -40,6 +40,7 @@ class ScoreModel(pl.LightningModule):
         """
         super().__init__()
         # Initialize Backbone DNN
+        self.backbone = backbone
         dnn_cls = BackboneRegistry.get_by_name(backbone)
         self.dnn = dnn_cls(**kwargs)
         # Initialize SDE