Index

_

__init__() (haiku_trainer.Trainer method)

A

aux (haiku_trainer.TrainerState attribute)

avg_training_loss() (haiku_trainer.Trainer method) avg_validation_loss() (haiku_trainer.Trainer method)

C

compile() (haiku_trainer.Trainer method)

F

find_latest_checkpoint() (haiku_trainer.Trainer method)

fit() (haiku_trainer.Trainer method)

L

load_state() (haiku_trainer.Trainer method)

load_step() (haiku_trainer.Trainer method)

O

optim (haiku_trainer.TrainerState attribute)

P

params (haiku_trainer.TrainerState attribute)

R

register_callback() (haiku_trainer.Trainer method) resume() (haiku_trainer.Trainer method)

rng (haiku_trainer.TrainerState attribute) run_func_with_state() (haiku_trainer.Trainer method)

S

save_state() (haiku_trainer.Trainer method)

save_step() (haiku_trainer.Trainer method)

T

tiktok() (haiku_trainer.Trainer method) Trainer (class in haiku_trainer)

TrainerState (class in haiku_trainer) training_step() (haiku_trainer.Trainer method) trange() (haiku_trainer.Trainer method)

V

validation_step() (haiku_trainer.Trainer method)