scico.flax.train

Utilities for training Flax models.

Modules

scico.flax.train.apply	Functionality to evaluate Flax trained model.
scico.flax.train.checkpoints	Utilities for checkpointing Flax models.
scico.flax.train.clu_utils	Utilities for displaying Flax models.
scico.flax.train.diagnostics	Utilities for computing and displaying performance metrics during training.
scico.flax.train.input_pipeline	Generalized data handling for training script.
scico.flax.train.learning_rate	Learning rate schedulers.
scico.flax.train.losses	Definition of loss functions for model optimization.
scico.flax.train.spectral	Utils for spectral normalization of convolutional layers in Flax models.
scico.flax.train.state	Configuration of Flax Train State.
scico.flax.train.steps	Definition of steps to iterate during training or evaluation.
scico.flax.train.trainer	Class providing integrated access to functionality for training Flax
scico.flax.train.traversals	Functionality to traverse, select, and update model parameters.
scico.flax.train.typed_dict	Definition of typed dictionaries for objects in training functionality.