echelon3

Describe a PyTorch training run in YAML — network, data, augmentations, losses, metrics, optimizer, scheduler, trainer, export — and run it with one command.

No training-loop boilerplate and no component registry: every piece is a module / type / config triple that echelon3 imports and instantiates, so anything importable drops straight in — classes from echelon3, from torch / timm / torchmetrics / albumentations, or from your own project code sitting next to your configs.

# my_run.yaml — a whole training run
net:  { module: my_pkg.nets, type: MyNet, config: { channels: 32 } }
data:
  train: { module: my_pkg.data, type: MyDataset, config: { split: train } }
  test:  { module: my_pkg.data, type: MyDataset, config: { split: test } }
dataloaders:
  train: { module: torch.utils.data, type: DataLoader, config: { batch_size: 64, shuffle: true } }
  test:  { module: torch.utils.data, type: DataLoader, config: { batch_size: 128 } }
loss:
  - ce: { module: torch.nn, type: CrossEntropyLoss, config: {} }
optimizer: { module: torch.optim, type: AdamW, config: { lr: 3e-4 } }
trainer:   { module: echelon3.trainers.baseline, type: Trainer, config: { epochs: 50 } }
target:    { path: ./out }

echelon3 train -cd . -cn my_run     # trains, validates, keeps the best, checkpoints, resumes

What you get

  • One CLI, five tasks — echelon3 train | finetune | evaluate | run | export.
  • Multi-GPU with no torchrun — gpus=[0,1,2,3] spawns one DDP worker per GPU; batch_size is global and split across ranks automatically.
  • bf16 mixed precision by default on capable GPUs (precision: fp32 to opt out), plus an optional torch.compile knob and TF32.
  • Batteries included in the trainer — automatic resume, keep-best-checkpoint, validation schedule, TensorBoard / mlops logging, clean Ctrl-C.
  • Override anything on the CLI — trainer.config.epochs=100 optimizer.config.lr=1e-4, ~scheduler drops a section, +key=… adds one; typed values, list literals, ${oc.env:VAR,default} interpolation, and defaults: config composition (OmegaConf).
  • Your own code, no registration — reference nets / datasets / losses by import path and run from your repo root (the cwd is on sys.path).

Install

pip install echelon3
# extras: echelon3[export] (ONNX)   echelon3[detection]   echelon3[smp]

CLI & overrides

echelon3 train    -cd ./configs -cn my_run          # train (auto-resumes from target.path)
echelon3 finetune -cd ./configs -cn my_run          # + warm-start / freeze / head-only
echelon3 evaluate -cd ./configs -cn my_run          # score the latest checkpoint
echelon3 export   -cd ./configs -cn my_run          # preprocess→net→postprocess to ONNX
echelon3 run      -cd ./configs -cn my_run          # inference over images / video

# override any config value; +add, ~delete, typed values, multi-GPU:
echelon3 train -cd ./configs -cn my_run \
    trainer.config.epochs=100 optimizer.config.lr=5e-4 \
    dataloaders.train.config.batch_size=256 \
    gpus=[0,1] +trainer.config.compile=true ~scheduler

--config-dir/-cd picks the directory, --config-name/-cn the YAML.

Quick start

examples/ has self-contained smoke runs — a classifier, a CenterNet-style detector and semantic segmentation — each with a synthetic-data generator and a minimal config that trains, validates and checkpoints on CPU or GPU. For example:

cd examples/segmentation
python gen_seg_data.py --root ./seg_data
SEG_DATA=./seg_data echelon3 train -cd . -cn segmentation_smoke device=cpu

Use it from your AI coding agent

Marketplace-installable plugins teach Codex and Claude Code the echelon3 config format and CLI, so the agent writes correct configs and runs them for you — repo: https://github.com/veryviolet/echelon3-agent-skills.

# Codex
codex plugin marketplace add veryviolet/echelon3-agent-skills
codex plugin add echelon3@veryviolet

# Claude Code (run inside the REPL)
/plugin marketplace add veryviolet/echelon3-agent-skills
/plugin install echelon3@veryviolet

Full documentation

https://veryviolet.github.io/echelon3/

Project links

Data verified by PyPI on Jul 27, 2026
Data provided by the project maintainers, verified at the time the release was uploaded to PyPI.

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Newer release available (0.11.0)

1 maintainer

PyPI data
Data sourced directly from PyPI's database.
Avatar for veryviolet from gravatar.com veryviolet

Credits

Author: Kirill Groshenkov

GitHub Statistics

Data verified by PyPI on Jul 27, 2026
The GitHub source repository was provided by the project maintainers and verified by PyPI at the time of upload. Stars, forks, and open issues/PRs are derived from that repository and have not been independently verified.

License expression

Apache-2.0
View SPDX License List

Requires

Python >=3.10

Provides Extra

detection docs export sam smp test

Tags

computer-vision config-driven deep-learning pytorch training

Classifiers

Report project as malware

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

echelon3-0.10.3.tar.gz (126.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

echelon3-0.10.3-py3-none-any.whl (169.2 kB view details)

Uploaded Python 3

File details

Details for the file echelon3-0.10.3.tar.gz.

File metadata

  • Download URL: echelon3-0.10.3.tar.gz
  • Upload date:
  • Size: 126.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for echelon3-0.10.3.tar.gz
Algorithm Hash digest
SHA256 14960ee37a635104c39a3c69aa0eb611e9c418aa0be3bd67db91e37607667c78
MD5 d6c9cd010f0abfae2e842cec8ad7044d
BLAKE2b-256 8af1a062bee73cee05358598f545edaaaa0e4dca342d8962edaa2fc3d83ef96a

See more details on using hashes here.

Provenance

The following attestation bundles were made for echelon3-0.10.3.tar.gz:

Publisher: publish.yml on veryviolet/echelon3

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file echelon3-0.10.3-py3-none-any.whl.

File metadata

  • Download URL: echelon3-0.10.3-py3-none-any.whl
  • Upload date:
  • Size: 169.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.14

File hashes

Hashes for echelon3-0.10.3-py3-none-any.whl
Algorithm Hash digest
SHA256 2b3b7770aaff20ce7023e47b1e1d978d67d45d3e5a037f36631d8fb6d8f950a8
MD5 c534e1b87ed63a3a202ad5726fb7153c
BLAKE2b-256 309fa3aa3e3ca8a84dd2448aef6e6691f0b53e6985322612d47bc5d050c7b5a9

See more details on using hashes here.

Provenance

The following attestation bundles were made for echelon3-0.10.3-py3-none-any.whl:

Publisher: publish.yml on veryviolet/echelon3

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

0.11.0

2 files

0.10.9

2 files

0.10.8

2 files

0.10.7

2 files

0.10.6

2 files

0.10.5

2 files

0.10.4

2 files

This release

0.10.3 This release

2 files

0.10.2

2 files

0.10.1

2 files

0.10.0

2 files

0.9.6

2 files

0.9.5

2 files

0.9.4

2 files

0.9.3

2 files

0.9.2

2 files

0.9.1

2 files

0.9.0

2 files

0.8.7

2 files

0.8.6

2 files

0.8.5

2 files

0.8.4

2 files

0.8.3

2 files

0.8.2

2 files

0.8.1

2 files

0.8.0

2 files

0.7.12

2 files

0.7.11

2 files

0.7.10

2 files

0.7.9

2 files

0.7.8

2 files

0.7.7

2 files

0.7.6

2 files

0.7.5

2 files

0.7.4

2 files

0.7.3

2 files

0.7.2

2 files

0.7.1

2 files

0.7.0

2 files

0.6.0

2 files

0.5.2

2 files

0.5.1

2 files

0.5.0

2 files

0.4.1

2 files

0.4.0

2 files

0.3.1

2 files

0.3.0

2 files

0.2.0

2 files

0.1.0

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page