hotfix optimal transport flow matching with conditions

bayesflow-org · Sep 9, 2024 · e5a78c4 · e5a78c4
1 parent e167ddd
commit e5a78c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bayesflow/networks/flow_matching/flow_matching.py b/bayesflow/networks/flow_matching/flow_matching.py
@@ -101,7 +101,7 @@ def compute_metrics(
 
             # use weak regularization and low number of steps for efficiency
             if self.use_optimal_transport:
-                x0, x1 = optimal_transport(x0, x1, seed=self.seed_generator, **self.optimal_transport_kwargs)
+                x1, x0 = optimal_transport(x1, x0, seed=self.seed_generator, **self.optimal_transport_kwargs)
 
             t = keras.random.uniform((keras.ops.shape(x0)[0],), seed=self.seed_generator)
             t = expand_right_as(t, x0)