ott.neural.solvers.neuraldual.W2NeuralDual.get_step_fn

ott.neural.solvers.neuraldual.W2NeuralDual.get_step_fn#

W2NeuralDual.get_step_fn(train, to_optimize)[source]#

Create a parallel training and evaluation function.

Parameters:
  • train (bool) –

  • to_optimize (Literal['f', 'g', 'parallel', 'both']) –