vis4d.engine
Vis4D run package.
Modules
Callback modules. |
|
Data connector for data connection. |
|
Data module composing the data loading pipeline. |
|
Engine Flags. |
|
Loss module maps loss function input keys and controls loss weight. |
|
Optimizer modules. |
|
Parser for config files that can be used with absl flags. |
|
CLI interface using PyTorch Lightning. |
|
Trainer for PyTorch Lightning. |
|
LightningModule that wraps around the models, losses and optims. |