scico.flax.train.typed_dict

Definition of typed dictionaries for objects in training functionality.

Classes

DataSetDict(*args, **kwargs)	Dictionary structure for training data sets.
MetricsDict(*args, **kwargs)	Dictionary structure for training metrics.
ModelVarDict(*args, **kwargs)	Dictionary structure for Flax variables.

class scico.flax.train.typed_dict.DataSetDict(*args, **kwargs)

Bases: dict

Dictionary structure for training data sets.

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

class scico.flax.train.typed_dict.ModelVarDict(*args, **kwargs)

Bases: dict

Dictionary structure for Flax variables.

Definition of the dictionary structure grouping all Flax model variables.

class scico.flax.train.typed_dict.MetricsDict(*args, **kwargs)

Bases: dict

Dictionary structure for training metrics.

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