ott.solvers.nn.neuraldual.NeuralDualSolver.setup#

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

Setup all components required to train the network.

Return type

None

Parameters