Integrations

Keras VisinCallback

VisinCallback(run: Run | None = None, *, first_epoch: int = 1, **init_kwargs: Any)

Report each epoch of training.

Pass run to report into a run you made. Otherwise one is started with :func:visin.init when training begins, given any other keyword arguments (name, project, tags...), and finished when training ends. Epochs are numbered from first_epoch, 1 by default, as Keras prints them. fit(initial_epoch=...) carries on from there. A run this callback made is finished when training ends, so calling fit again starts a new one.

Lightning VisinCallback

VisinCallback(run: Run | None = None, *, first_epoch: int = 1, finish_after: str = 'fit', log_hyperparameters: bool = True, **init_kwargs: Any)

Report each training epoch, and test results.

Pass run to report into a run you made. Otherwise one is started with :func:visin.init at setup, given any other keyword arguments (name, project, tags...), and finished after finish_after: "fit" by default, or "test" when trainer.test follows trainer.fit and its results should reach the run too.

Hugging Face VisinCallback

VisinCallback(run: Run | None = None, **init_kwargs: Any)

Report each epoch of Trainer training.

Pass run to report into a run you made. Otherwise one is started with :func:visin.init when training begins, given any other keyword arguments (name, project, tags...), with the TrainingArguments as its config, and finished when training ends, so a second train starts a new run.