Adding Things¶
Theseus is structured around four extension points. Each one follows the same pattern: write a class, drop it in the right folder, register it.
| What | Where | How |
|---|---|---|
| Model | theseus/model/models/ |
subclass Module, add to __init__.py |
| Experiment | theseus/experiments/ |
@job("key") + subclass BaseTrainer |
| Dataset | theseus/data/datasets/ |
@dataset("key") + subclass a dataset base |
| Evaluation | theseus/evaluation/datasets/ |
@evaluation("key") + subclass RolloutEvaluation |