Added GINCouplingBlock to __init__.py

vislearn · Nov 12, 2020 · c5fe1af · c5fe1af
1 parent a09fe7a
commit c5fe1af
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/FrEIA/modules/__init__.py b/FrEIA/modules/__init__.py
@@ -8,6 +8,7 @@
 * NICECouplingBlock
 * RNVPCouplingBlock
 * GLOWCouplingBlock
+* GINCouplingBlock
 * AffineCouplingOneSided
 * ConditionalAffineTransform
 
@@ -94,6 +95,7 @@
             'NICECouplingBlock',
             'RNVPCouplingBlock',
             'GLOWCouplingBlock',
+            'GINCouplingBlock',
             'AffineCouplingOneSided',
             'ConditionalAffineTransform',
             'PermuteRandom',