vis4d.engine

Vis4D run package.

Modules

callbacks

Callback modules.

connectors

Data connector for data connection.

data_module

Data module composing the data loading pipeline.

flag

Engine Flags.

loss_module

Loss module maps loss function input keys and controls loss weight.

optim

Optimizer modules.

parser

Parser for config files that can be used with absl flags.

run

CLI interface using PyTorch Lightning.

trainer

Trainer for PyTorch Lightning.

training_module

LightningModule that wraps around the models, losses and optims.