ott.core.neuraldual.NeuralDualSolver.get_step_fn#

NeuralDualSolver.get_step_fn(train, to_optimize='g')[source]#

Create a one-step training and evaluation function.