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