Add sigmoid-based multiscale architectures to architectures.py

davidnabergoj · Jul 6, 2024 · d42ac60 · d42ac60
1 parent 727b4b5
commit d42ac60
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/normalizing_flows/architectures.py b/normalizing_flows/architectures.py
@@ -30,5 +30,8 @@
     MultiscaleRealNVP,
     MultiscaleRQNSF,
     MultiscaleLRSNSF,
-    MultiscaleNICE
+    MultiscaleNICE,
+    # MultiscaleDeepSigmoid,  # TODO stabler initialization
+    # MultiscaleDenseSigmoid,  # TODO stabler initialization
+    # MultiscaleDeepDenseSigmoid  # TODO stabler initialization
 )