vis4d.engine.callbacks.util

PyTorch Lightning callbacks utilities.

Functions

`get_loss_module`(loss_module)	Get loss_module from pl module.
`get_model`(model)	Get model from pl module.

get_model(model)[source]

Get model from pl module.

get_loss_module(loss_module)[source]

Get loss_module from pl module.