vis4d.plΒΆ
Run training and evaluation in PyTorch Lightening.
Modules
Callbacks handling data related stuff (evaluation, visualization, etc). |
|
Data module composing the data loading pipeline. |
|
CLI interface using PyTorch Lightning. |
|
Trainer for PyTorch Lightning. |
|
LightningModule that wraps around the models, losses and optims. |