Add autoregressive NAF architectures to __init__.py

davidnabergoj · Aug 23, 2024 · b779377 · b779377
1 parent 0b7b42d
commit b779377
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/torchflows/architectures.py b/torchflows/architectures.py
@@ -11,8 +11,14 @@
     MaskedAutoregressiveLRS,
     InverseAutoregressiveLRS,
     CouplingDeepSF,
+    MaskedAutoregressiveDeepSF,
+    InverseAutoregressiveDeepSF,
     CouplingDenseSF,
+    MaskedAutoregressiveDenseSF,
+    InverseAutoregressiveDenseSF,
     CouplingDeepDenseSF,
+    MaskedAutoregressiveDeepDenseSF,
+    InverseAutoregressiveDeepDenseSF,
     UMNNMAF
 )