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]]
