[BUG] Remove res["pi"].detach().cpu()

[pbarbarant]

In fugw.mapping :

# Store variables of interest in model
self.pi = res["pi"].detach().cpu()

Causes problems when dealing with barycenters as the transport plans get detached before the next barycenter iteration. Discovered with @antoinecollas.