ott.solvers.nn.neuraldual.NeuralDualSolver.get_step_fn

NeuralDualSolver.get_step_fn(train, to_optimize='g')

Create a one-step training and evaluation function.

Parameters

• train (bool) –

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