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)