wncc/UniTrain
UniTrain is an open-source, unified platform for effortless machine learning model training, evaluation, and deployment across diverse tasks. Experience seamless experimentation and model deployment with UniTrain.
This platform helps machine learning engineers efficiently train, evaluate, and deploy various models without writing extensive boilerplate code. You provide structured image datasets for tasks like classification, segmentation, or generative modeling, and it outputs trained models ready for use. It's designed for data scientists and ML engineers building custom computer vision solutions.
No commits in the last 6 months.
Use this if you are an ML engineer or data scientist who needs a streamlined way to train, test, and prepare computer vision models for deployment across different project types.
Not ideal if you are looking for a no-code solution or a tool for general machine learning tasks outside of computer vision.
Stars
18
Forks
23
Language
Python
License
MIT
Category
Last pushed
Oct 31, 2023
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/wncc/UniTrain"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
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, ...)