verta_hook

verta.integrations.torch.verta_hook(run)

PyTorch module hook that automates logging to Verta during training.

This hook logs details about the network topology.

See our GitHub repository for an example of this intergation in action.

New in version 0.13.20.

Parameters:

run (ExperimentRun) – Experiment Run tracking this model.

Examples

from verta.integrations.torch import verta_hook
run = client.set_experiment_run()
model.register_forward_hook(verta_hook(run))
output = model(X_train)