mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-07-08 05:10:09 +02:00
67aee0b79c
Some checks failed
CI / Checks (push) Has been cancelled
Docs Pages / Build Docs (push) Has been cancelled
CI / Tests (Python ${{ matrix.python-version }}) (3.10) (push) Has been cancelled
CI / Tests (Python ${{ matrix.python-version }}) (3.11) (push) Has been cancelled
CI / Tests (Python ${{ matrix.python-version }}) (3.12) (push) Has been cancelled
Docs Pages / Deploy Docs (push) Has been cancelled
51 lines
1.3 KiB
Markdown
51 lines
1.3 KiB
Markdown
# Training config reference
|
|
|
|
`TrainingConfig` controls the training loop, optimisation, data loading, losses,
|
|
and validation tasks.
|
|
|
|
Defined in `batdetect2.train.config`.
|
|
|
|
## Top-level fields
|
|
|
|
- `train_loader`
|
|
- training data loading and clipping settings.
|
|
- `val_loader`
|
|
- validation data loading and clipping settings.
|
|
- `optimizer`
|
|
- optimiser type and learning rate settings.
|
|
- `scheduler`
|
|
- learning-rate schedule settings.
|
|
- `loss`
|
|
- detection, classification, and size loss settings.
|
|
- `trainer`
|
|
- PyTorch Lightning trainer settings such as `max_epochs`.
|
|
- `labels`
|
|
- target label generation settings.
|
|
- `validation`
|
|
- evaluation tasks used during validation.
|
|
- `checkpoints`
|
|
- checkpoint saving settings.
|
|
|
|
## What this config controls
|
|
|
|
Use `TrainingConfig` when you want to change things like:
|
|
|
|
- batch size,
|
|
- augmentation,
|
|
- optimiser and scheduler settings,
|
|
- number of epochs,
|
|
- validation frequency,
|
|
- checkpoint behaviour.
|
|
|
|
Example files live under `example_data/configs/`, including
|
|
`example_data/configs/training.yaml`.
|
|
|
|
## Related pages
|
|
|
|
- Evaluation config:
|
|
{doc}`../evaluation/evaluation-config`
|
|
- Train command reference:
|
|
{doc}`../../cli/train`
|
|
- Fine-tune from a checkpoint:
|
|
{doc}`../../../how_to/training/fine-tune-from-a-checkpoint`
|