ott.core.neuraldual.NeuralDualSolver.setup#

NeuralDualSolver.setup(rng, neural_f, neural_g, input_dim, optimizer_f, optimizer_g)[source]#

Setup all components required to train the NeuralDual.