ott.solvers.nn.models.ICNN.create_train_state#

ICNN.create_train_state(rng, optimizer, input, **kwargs)#

Create initial training state.

Parameters:
Return type:

NeuralTrainState