From 089a625f0238e7b0f48eef0fb5154747671395c6 Mon Sep 17 00:00:00 2001
From: jrichter <jrichter@exchange.informatik.uni-hamburg.de>
Date: Sun, 16 Jun 2024 14:21:25 +0200
Subject: [PATCH] add backbone parameter
---
sgmse/model.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/sgmse/model.py b/sgmse/model.py
index 2baed0e..a97a566 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
--
GitLab