Callbacks

Training lifecycle callbacks and reporting integrations.

class taktiny.trainer.TrainerCallback[source]

Bases: object

Base class for Trainer lifecycle callbacks.

class taktiny.trainer.TensorBoardCallback(log_dir=None, *, writer=None)[source]

Bases: TrainerCallback

Write Trainer logs through a TensorBoard SummaryWriter.

Parameters:

  • log_dir (Any)

  • writer (Any)

class taktiny.trainer.WandbCallback(project=None, *, name=None, config=None, run=None, **init_kwargs)[source]

Bases: TrainerCallback

Report Trainer logs to Weights & Biases.

Parameters:

  • project (Any)

  • name (str | None)

  • config (Any)

  • run (Any)

  • init_kwargs (Any)