ott.neural.methods.expectile_neural_dual.ExpectileNeuralDual.train_fn

ott.neural.methods.expectile_neural_dual.ExpectileNeuralDual.train_fn#

ExpectileNeuralDual.train_fn(trainloader_source, trainloader_target, validloader_source, validloader_target, callback=None)[source]#

Training and validation.

Parameters:
Return type:

Dict[Literal['train_logs', 'valid_logs'], Dict[str, List[float]]]