Skip to content
Snippets Groups Projects
Commit 089a625f authored by jrichter's avatar jrichter
Browse files

add backbone parameter

parent f0a91d59
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment