Disable mvscale in coupling

aseyboldt · aseyboldt · commit 5dcda60f3c21 · 2025-02-20T17:38:08.000+01:00
diff --git a/python/nutpie/normalizing_flow.py b/python/nutpie/normalizing_flow.py
@@ -514,7 +514,7 @@ def make_coupling(key, dim, n_untransformed, **kwargs):
     transformer = bijections.Chain(
         [
             make_elemwise_trafo(key, n_transformed, count=3),
-            mvscale,
+            #mvscale,
         ]
     )
 

Original file line number	Diff line number	Diff line change
`@@ -514,7 +514,7 @@ def make_coupling(key, dim, n_untransformed, **kwargs):`
`514`	`514`	`transformer = bijections.Chain(`
`515`	`515`	`[`
`516`	`516`	`make_elemwise_trafo(key, n_transformed, count=3),`
`517`		`- mvscale,`
	`517`	`+ #mvscale,`
`518`	`518`	`]`
`519`	`519`	`)`
`520`	`520`