Trainers registry#
Trainer classes selected by train.type. See Architecture and extension seams.
Build any of these from a config with train = dict(type=...), using the type string in the first column. Generated from the live registry - 5 classes (5 names including aliases), grouped by role.
General#
The default training loop and multi-dataset variant.
Name |
Summary |
|---|---|
Default single-dataset supervised/SSL trainer. |
|
Trainer that draws mixed batches from several datasets. |
Default single-dataset supervised/SSL trainer. |
|
Trainer that draws mixed batches from several datasets. |
Task-specific#
Trainers with task-specific collation / loaders.
Name |
Summary |
|---|---|
Trainer for dense 2D image batches (e.g. rasterized ring images). |
|
Trainer for instance segmentation with instance-aware collation. |
Trainer for dense 2D image batches (e.g. rasterized ring images). |
|
Trainer for instance segmentation with instance-aware collation. |
Reinforcement learning#
Rollout-based policy optimization (GRPO).
Name |
Summary |
|---|---|
Reinforcement-learning trainer implementing GRPO at the trainer level. |
Reinforcement-learning trainer implementing GRPO at the trainer level. |