ott.neural.methods.monge_gap.MongeGapEstimator.train_map_estimator# MongeGapEstimator.train_map_estimator(trainloader_source, trainloader_target, validloader_source, validloader_target)[source]# Training loop. Parameters: trainloader_source (Iterator[Array]) – trainloader_target (Iterator[Array]) – validloader_source (Iterator[Array]) – validloader_target (Iterator[Array]) – Return type: Tuple[TrainState, Dict[str, Any]]