ott.solvers.nn.neuraldual.W2NeuralDual.train_neuraldual_parallel# W2NeuralDual.train_neuraldual_parallel(trainloader_source, trainloader_target, validloader_source, validloader_target, callback=None)[source]# Training and validation with parallel updates. Parameters: trainloader_source (Iterator[Array]) – trainloader_target (Iterator[Array]) – validloader_source (Iterator[Array]) – validloader_target (Iterator[Array]) – callback (Optional[Callable[[int, DualPotentials], None]]) – Return type: Dict[Literal['train_logs', 'valid_logs'], Dict[str, List[float]]]
