ott.solvers.nn.neuraldual.W2NeuralDual.get_step_fn

W2NeuralDual.get_step_fn(train, to_optimize)

Create a parallel training and evaluation function.

Parameters:

• train (bool) –

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