ott.neural.solvers.neuraldual.W2NeuralDual.to_dual_potentials

ott.neural.solvers.neuraldual.W2NeuralDual.to_dual_potentials#

W2NeuralDual.to_dual_potentials(finetune_g=True)[source]#

Return the Kantorovich dual potentials from the trained potentials.

Parameters:

finetune_g (bool) – Run the conjugate solver to fine-tune the prediction.

Return type:

DualPotentials

Returns:

A dual potential object