scico.flax.train.typed_dict

Definition of typed dictionaries for objects in training functionality.

Classes

DataSetDict	Dictionary structure for training data sets.
MetricsDict	Dictionary structure for training metrics.
ModelVarDict	Dictionary structure for Flax variables.

class scico.flax.train.typed_dict.DataSetDict

Bases: TypedDict

Dictionary structure for training data sets.

Definition of the dictionary structure expected for the training data sets.

class scico.flax.train.typed_dict.ModelVarDict

Bases: TypedDict

Dictionary structure for Flax variables.

Definition of the dictionary structure grouping all Flax model variables.

class scico.flax.train.typed_dict.MetricsDict

Bases: TypedDict

Dictionary structure for training metrics.

Definition of the dictionary structure for metrics computed or updates made during training.