echelon3

Config-driven PyTorch training framework. Every component of a training run — network, dataset, augmentations, losses, metrics, optimizer, scheduler, trainer, export — is described in a YAML config as a module / type / config triple and instantiated dynamically:

net:
  module: echelon3.nets.classifier   # import path (or a path to a .py file)
  type: ClassifierNet                # class or factory function in that module
  config: { ... }                    # constructor kwargs

There is no component registry: anything importable can be plugged in — classes from echelon3, from torch/timm/torchmetrics/albumentations, or from your own project code living next to your configs.

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

Install

pip install echelon3

Train

echelon3-train --config-dir ./configs --config-name my_experiment

CLIs: echelon3-train, echelon3-finetune (warm-start / freeze / head-only), echelon3-evaluate, echelon3-run (inference), echelon3-export (ONNX).

Multi-GPU — built in, no torchrun

Name the GPUs and echelon3 spawns one DDP worker per GPU itself:

echelon3-train --config-dir ./configs --config-name my_experiment gpus=[0,1,2,3]

gpus is a root config key — leave it out and echelon3 uses every visible GPU on the node. dataloaders.train.config.batch_size is the global batch size; it is split across ranks automatically. torchrun (and SLURM srun) still work unchanged for multi-node / elastic jobs.

DataParallel was removed in 0.5.0 — multiple GPUs always run as DDP.

Mixed precision

Training, evaluation and inference use bf16 automatic mixed precision by default on capable GPUs (fp32 on CPU / unsupported GPUs) — a large speedup on modern hardware. Force full fp32 with precision: fp32 under trainer.config (or precision: fp32 at the config root for evaluate / run).

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.

Project links

Data verified by PyPI on Jul 8, 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 8, 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 hydra 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.7.3.tar.gz (127.6 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.7.3-py3-none-any.whl (130.5 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for echelon3-0.7.3.tar.gz
Algorithm Hash digest
SHA256 1ae34ce68d5d6d5541e2292904cbba92859dffe9e93993be63183cc64cacecb6
MD5 8a6aedb6d1cf855184fa7ddd19fab436
BLAKE2b-256 46c9002732f0fcf67fad3c1c7aa048f4cc4cf596d0d296df42a0ec3559f73274

See more details on using hashes here.

Provenance

The following attestation bundles were made for echelon3-0.7.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.7.3-py3-none-any.whl.

File metadata

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

File hashes

Hashes for echelon3-0.7.3-py3-none-any.whl
Algorithm Hash digest
SHA256 789a71ccf39ccd2ed69f4a937d6f74249881f3b9628c952298ba8f3113f964ca
MD5 30e29573c621b35c8b56db73e825c74a
BLAKE2b-256 a02236b9d5ed0b68a0a9ea4284f01b01bdea0b50cbbced880d1afa144944e230

See more details on using hashes here.

Provenance

The following attestation bundles were made for echelon3-0.7.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

0.10.3

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

This release

0.7.3 This release

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