Adds network and tests

Project-MONAI · Nov 27, 2023 · d99044b · d99044b
1 parent 8e134b8
commit d99044b
Show file tree

Hide file tree

Showing 4 changed files with 1,092 additions and 0 deletions.
diff --git a/docs/source/networks.rst b/docs/source/networks.rst
@@ -595,6 +595,11 @@ Nets
 .. autoclass:: AutoEncoder
   :members:
 
+`AutoEncoderKL`
+~~~~~~~~~~~~~
+.. autoclass:: AutoEncoderKL
+  :members:
+
 `VarAutoEncoder`
 ~~~~~~~~~~~~~~~~
 .. autoclass:: VarAutoEncoder

diff --git a/monai/networks/nets/__init__.py b/monai/networks/nets/__init__.py
@@ -14,6 +14,7 @@
 from .ahnet import AHnet, Ahnet, AHNet
 from .attentionunet import AttentionUnet
 from .autoencoder import AutoEncoder
+from .autoencoderkl import AutoencoderKL
 from .basic_unet import BasicUNet, BasicUnet, Basicunet, basicunet
 from .basic_unetplusplus import BasicUNetPlusPlus, BasicUnetPlusPlus, BasicunetPlusPlus, basicunetplusplus
 from .classifier import Classifier, Critic, Discriminator