ott.neural.methods.monge_gap.MongeGapEstimator.setup# MongeGapEstimator.setup(dim_data, neural_net, optimizer)[source]# Setup all components required to train the network. Parameters: dim_data (int) neural_net (BasePotential) optimizer (Union[Array, ndarray, bool, number, Iterable[ArrayTree], Mapping[Any, ArrayTree]])