cellflow.networks.ConditionEncoder.create_train_state

ConditionEncoder.create_train_state(rng, optimizer, conditions, **kwargs)

Create initial training state.

Parameters:

• rng (Array)

• optimizer (Array | ndarray | bool | number | bool | int | float | complex | Iterable[ArrayTree] | Mapping[Any, ArrayTree])

• conditions (dict[str, Array])

• kwargs (Any)