training

sleap_nn.training

Training-related modules.

Modules:

Name	Description
callbacks	Custom Callback modules for Lightning Trainer.
lightning_modules	This module has the LightningModule classes for all model types.
losses	Custom loss functions.
model_trainer	This module is to train a sleap-nn model using Lightning.
schedulers	Custom learning rate schedulers for sleap-nn training.
utils	Miscellaneous utility functions for training.