add bayesian vnet to list of available models

neuronets · May 21, 2024 · a8e960b · a8e960b
1 parent d01e48a
commit a8e960b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nobrainer/models/__init__.py b/nobrainer/models/__init__.py
@@ -4,6 +4,7 @@
 from .attention_unet_with_inception import attention_unet_with_inception
 from .autoencoder import autoencoder
 from .bayesian_meshnet import variational_meshnet
+from .bayesian_vnet import bayesian_vnet
 from .dcgan import dcgan
 from .highresnet import highresnet
 from .meshnet import meshnet
@@ -26,6 +27,7 @@
     "attention_unet_with_inception": attention_unet_with_inception,
     "unetr": unetr,
     "variational_meshnet": variational_meshnet,
+    "bayesian_vnet": bayesian_vnet
 }