ott.neural.methods.neuraldual.W2NeuralDual.setup

W2NeuralDual.setup(neural_f, neural_g, optimizer_f, optimizer_g)

Setup all components required to train the network.

Parameters:

• neural_f (Module)

• neural_g (Module)

• optimizer_f (Union[Array, ndarray, bool, number, bool, int, float, complex, Iterable[ArrayTree], Mapping[Any, ArrayTree]])

• optimizer_g (Union[Array, ndarray, bool, number, bool, int, float, complex, Iterable[ArrayTree], Mapping[Any, ArrayTree]])

Return type:

None