ott.neural.methods.neuraldual.W2NeuralDual.get_step_fn# W2NeuralDual.get_step_fn(train, to_optimize)[source]# Create a parallel training and evaluation function. Parameters: train (bool) to_optimize (Literal['f', 'g', 'parallel', 'both'])