coqui-ai/Trainer
🐸 - A general purpose model trainer, as flexible as it gets
This is a versatile tool for machine learning engineers and researchers to train deep learning models on PyTorch efficiently. It streamlines the process of taking your model and data, applying various optimization strategies, and producing a trained model. It's designed for users who need fine-grained control over their training loops or want to leverage advanced features like distributed training and batch size optimization.
233 stars. Used by 1 other package. No commits in the last 6 months. Available on PyPI.
Use this if you are a machine learning engineer or researcher developing deep learning models in PyTorch and need a flexible, controlled environment for training, optimization, and experimentation.
Not ideal if you are looking for a high-level, 'one-click' solution for standard machine learning tasks without needing to customize training specifics or if you're not comfortable working with PyTorch code.
Stars
233
Forks
148
Language
Python
License
—
Category
Last pushed
Mar 07, 2024
Commits (30d)
0
Dependencies
6
Reverse dependents
1
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/coqui-ai/Trainer"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
pytorch/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras-team/keras
Deep Learning for humans
Lightning-AI/torchmetrics
Machine learning metrics for distributed, scalable PyTorch applications.
Lightning-AI/pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
lanpa/tensorboardX
tensorboard for pytorch (and chainer, mxnet, numpy, ...)