skillfed

coqui-tts-trainer

General purpose model trainer for PyTorch that is more flexible than it should be, by 🐸Coqui.

coqui-tts-trainer v0.4.0 146.9K downloads/30d#11,087 on PyPI16
Permissive license Apache-2.0 Active released

What it is and what it does

Coqui-tts-trainer is a PyTorch training framework that abstracts away boilerplate training loop code while preserving flexibility for custom optimization logic. It provides opinionated defaults for common tasks—auto-optimization, mixed precision, gradient accumulation, and distributed training via DDP or Hugging Face Accelerate—but allows you to override the optimization cycle entirely when needed. The framework integrates with multiple experiment loggers (TensorBoard, ClearML, MLflow, Aim, WandB) and includes utilities like batch size finder and profiling support.

You define your model by subclassing TrainerModel and overloading its methods, then pass it to a Trainer instance with configuration. The trainer handles the training loop, checkpointing, logging, and device management. It supports both simple auto-optimized training (useful for standard supervised learning) and fully custom optimization loops (useful for GANs and other adversarial setups), making it suitable for researchers and practitioners who want structure without sacrificing control.

Use it for:

  • Train standard supervised models with auto-optimization and minimal configuration overhead.
  • Implement GAN or multi-network training with custom per-step optimization logic and gradient accumulation.
  • Automatically find the largest batch size that fits on your hardware without manual tuning.
  • Distribute training across multiple GPUs using DDP or Accelerate without rewriting your model code.
  • Profile training performance and memory usage with integrated PyTorch profiler and TensorBoard visualization.
  • Log metrics and checkpoints to multiple experiment tracking platforms simultaneously.

Worth the install?

AI-flagged interpretation of the facts on this page — verify before relying

A PyTorch model trainer framework that handles training loops, optimization, mixed precision, distributed training, and experiment logging with minimal boilerplate.

Yes. The package is actively maintained, has low install friction, carries a permissive Apache-2.0 license, and solves a real problem—reducing boilerplate in PyTorch training loops while preserving flexibility. It is suitable for both simple supervised learning and advanced custom training scenarios. No known security vulnerabilities. Best for teams already committed to PyTorch who want a structured but not rigid training abstraction.

Install

coqui-tts-trainer on PyPI

pip

pip install coqui-tts-trainer

uv

uv add coqui-tts-trainer

poetry

poetry add coqui-tts-trainer

Installing coqui-tts-trainer

Before you install

Low install friction with a pure-Python wheel and five lightweight runtime dependencies. Actively maintained as of April 2026 with recent commits.

License in practice

Apache-2.0 permissive license allows commercial and private use with minimal restrictions.

Quickstart

pip install coqui-tts-trainer

from trainer import Trainer, TrainerModel

class MyModel(TrainerModel):
    def forward(self, batch):
        return self.model(batch)

trainer = Trainer(model=MyModel(), ...)
trainer.fit()

Requires PyTorch (not listed as explicit runtime dep but essential for the framework to function).

Verify before relying

  • Whether PyTorch is an implicit dependency or must be installed separately.
  • Performance characteristics and scalability limits for large-scale distributed training.
  • Compatibility with recent PyTorch versions beyond what the classifiers indicate.

Package facts

License Apache-2.0 (permissive)
Python support supports the current Python release (<3.15,>=3.10)
Install friction low — pure-Python wheel
Runtime dependencies 5 — coqpit-config, fsspec, packaging, psutil, tensorboard
Maintenance actively maintained — 126 days since the last release
Last repo commit
First released
Downloads 146,934/month — #11,087 on PyPI (30-day window, as of 2026-08-14)
Known vulnerabilities none known (OSV.dev, checked 2026-08-14)

Evidence: coqui_tts_trainer-0.4.0-py3-none-any.whl

Development Status :: 4 - BetaEnvironment :: ConsoleIntended Audience :: DevelopersIntended Audience :: Science/ResearchLicense :: OSI Approved :: Apache Software LicenseNatural Language :: EnglishOperating System :: OS IndependentProgramming Language :: PythonProgramming Language :: Python :: 3Programming Language :: Python :: 3.10Programming Language :: Python :: 3.11Programming Language :: Python :: 3.12Programming Language :: Python :: 3.13Programming Language :: Python :: 3.14Topic :: Scientific/Engineering :: Artificial IntelligenceTopic :: Software DevelopmentTopic :: Software Development :: Libraries :: Python Modules

Tags

pytorch model trainer frameworktraining loop abstractiondistributed training pytorchmixed precision trainingexperiment logging trainergan training frameworkbatch size finder
pytorch-trainingdistributed-trainingexperiment-logging

More Software Development packages

Further reading