From ccc076e0fc535b2c734214c0ac1936e9e2cbeb62 Mon Sep 17 00:00:00 2001
From: eloitanguy <69361683+eloitanguy@users.noreply.github.com>
Date: Fri, 6 May 2022 08:43:21 +0200
Subject: [WIP] Generalized Wasserstein Barycenters  (#372)

* GWB first solver version

* tests + example for gwb (untested) + free_bar doc fix

* improved doc, fixed minor bugs, better example visu

* minor doc + visu fixes

* plot GWB pep8 fix

* fixed partial gromov test reproductibility

* added an animation for the GWB visu

* added PR num

* minor doc fixes + better gwb logo
---
 examples/backends/plot_sliced_wass_grad_flow_pytorch.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'examples/backends/plot_sliced_wass_grad_flow_pytorch.py')

diff --git a/examples/backends/plot_sliced_wass_grad_flow_pytorch.py b/examples/backends/plot_sliced_wass_grad_flow_pytorch.py
index cf5d64d..59e0042 100644
--- a/examples/backends/plot_sliced_wass_grad_flow_pytorch.py
+++ b/examples/backends/plot_sliced_wass_grad_flow_pytorch.py
@@ -103,7 +103,7 @@ ax = pl.axis()
 
 # %%
 # Animate trajectories of the gradient flow along iteration
-# -------------------------------------------------------
+# ---------------------------------------------------------
 
 pl.figure(3, (8, 4))
 
@@ -122,7 +122,7 @@ ani = animation.FuncAnimation(pl.gcf(), _update_plot, nb_iter_max, interval=100,
 
 # %%
 # Compute the Sliced Wasserstein Barycenter
-#
+# -----------------------------------------
 x1_torch = torch.tensor(x1).to(device=device)
 x3_torch = torch.tensor(x3).to(device=device)
 xbinit = np.random.randn(500, 2) * 10 + 16
@@ -169,7 +169,7 @@ ax = pl.axis()
 
 # %%
 # Animate trajectories of the barycenter along gradient descent
-# -------------------------------------------------------
+# -------------------------------------------------------------
 
 pl.figure(5, (8, 4))
 
-- 
cgit v1.2.3