Index

B

batch_size (learnax.Trainer attribute) (learnax.trainer.Trainer attribute)

best_metric_value (learnax.checkpoint.Checkpointer attribute), [1]

C

checkpoint_dir (learnax.checkpoint.Checkpointer attribute), [1] checkpoint_every (learnax.checkpoint.Checkpointer attribute), [1] Checkpointer (class in learnax.checkpoint)

checkpointer (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) clear_dir() (learnax.run.Run method)

D

device_count (learnax.Trainer attribute) (learnax.trainer.Trainer attribute)

E

epoch() (learnax.Trainer method) (learnax.trainer.Trainer method)

F

fetch_run() (learnax.Registry method) (learnax.registry.Registry method)

find_class() (learnax.registry.RenameUnpickler method) (learnax.RenameUnpickler method)

G

get_config() (learnax.run.Run method) get_latest_step() (learnax.checkpoint.Checkpointer method) get_train_state() (learnax.run.Run method)

get_weights() (learnax.run.Run method) grad() (learnax.Trainer method) (learnax.trainer.Trainer method)

I

id (learnax.run.Run attribute) in_notebook() (in module learnax) (in module learnax.trainer)

init() (learnax.Trainer method) (learnax.trainer.Trainer method)

K

keep_best (learnax.checkpoint.Checkpointer attribute), [1]

L

learnax module learnax.checkpoint module learnax.loss module learnax.registry module learnax.run module learnax.sampler module learnax.trainer module learnax.utils module libcudart (in module learnax) (in module learnax.trainer)

list_available_checkpoints() (learnax.checkpoint.Checkpointer method) load_best() (learnax.checkpoint.Checkpointer method) load_checkpoint() (learnax.checkpoint.Checkpointer method) load_latest() (learnax.checkpoint.Checkpointer method) load_weights (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) loaders (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) log() (learnax.run.Run method) loss() (learnax.Trainer method) (learnax.trainer.Trainer method) loss_list (learnax.loss.LossPipe attribute) losses (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) LossFunction (class in learnax.loss) LossPipe (class in learnax.loss)

M

max_grad (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) max_to_keep (learnax.checkpoint.Checkpointer attribute), [1] maybe_save_checkpoints() (learnax.checkpoint.Checkpointer method) metric_name (learnax.checkpoint.Checkpointer attribute), [1] metrics (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) minimize_metric (learnax.checkpoint.Checkpointer attribute), [1] model (learnax.Trainer attribute) (learnax.trainer.Trainer attribute)

module learnax learnax.checkpoint learnax.loss learnax.registry learnax.run learnax.sampler learnax.trainer learnax.utils

N

name (learnax.run.Run attribute) (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) new() (learnax.run.Run class method) new_run() (learnax.Registry method) (learnax.registry.Registry method)

num_epochs (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) num_workers (learnax.Trainer attribute) (learnax.trainer.Trainer attribute)

O

opt_state (learnax.trainer.TrainState attribute) (learnax.TrainState attribute)

optimizer (learnax.Trainer attribute) (learnax.trainer.Trainer attribute)

P

params (learnax.trainer.TrainState attribute) (learnax.TrainState attribute) path (learnax.Registry attribute) (learnax.registry.Registry attribute) (learnax.run.Run attribute) plot_every (learnax.Trainer attribute) (learnax.trainer.Trainer attribute)

plot_model (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) plot_pipe (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) prepare_fn (learnax.checkpoint.Checkpointer attribute), [1] project (learnax.Registry attribute) (learnax.registry.Registry attribute)

R

read() (learnax.run.Run method) read_all() (learnax.run.Run method) Registry (class in learnax) (class in learnax.registry) registry_path (learnax.checkpoint.Checkpointer attribute), [1] renamed_load() (in module learnax) (in module learnax.registry) RenameUnpickler (class in learnax) (class in learnax.registry)

restore() (learnax.run.Run class method) restore_run() (learnax.Registry method) (learnax.registry.Registry method) Run (class in learnax.run) run (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) run_id (learnax.checkpoint.Checkpointer attribute), [1]

S

sample_every (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) save() (learnax.run.Run method) save_checkpoint() (learnax.checkpoint.Checkpointer method) save_every (learnax.checkpoint.Checkpointer attribute), [1] save_latest() (learnax.checkpoint.Checkpointer method) save_model (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) scheduler (learnax.loss.LossFunction attribute)

seed (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) should_save_latest() (learnax.checkpoint.Checkpointer method) should_save_numbered() (learnax.checkpoint.Checkpointer method) single_batch (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) single_datum (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) step (learnax.trainer.TrainState attribute) (learnax.TrainState attribute)

T

train() (learnax.Trainer method) (learnax.trainer.Trainer method) train_dataset (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) train_only (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) Trainer (class in learnax) (class in learnax.trainer)

TrainState (class in learnax) (class in learnax.trainer) transform (learnax.Trainer attribute) (learnax.trainer.Trainer attribute) tree_stack() (in module learnax), [1] (in module learnax.utils) tree_unstack() (in module learnax), [1] (in module learnax.utils)

U

update() (learnax.Trainer method) (learnax.trainer.Trainer method)

update_best_checkpoint() (learnax.checkpoint.Checkpointer method)

V

val_every (learnax.Trainer attribute) (learnax.trainer.Trainer attribute)

W

wandb_run (learnax.run.Run attribute)

weight (learnax.loss.LossFunction attribute)