ott.neural.solvers.neuraldual.W2NeuralDual.setup

ott.neural.solvers.neuraldual.W2NeuralDual.setup#

W2NeuralDual.setup(rng, neural_f, neural_g, dim_data, optimizer_f, optimizer_g)[source]#

Setup all components required to train the network.

Parameters:
Return type:

None