PanJinquan/Pytorch-Base-Trainer
Pytorch分布式训练框架
This project helps deep learning practitioners train their PyTorch models more efficiently, especially when using multiple GPUs. It provides a structured framework, similar to Keras, for defining training, testing, and optimization processes. You input your PyTorch model, datasets, and training configurations, and it outputs trained models ready for deployment, often with performance metrics visualized.
Use this if you are a deep learning engineer or researcher looking for a lightweight, easy-to-install PyTorch training framework that supports multi-GPU and distributed training out-of-the-box.
Not ideal if you prefer highly opinionated, large-scale training libraries with extensive pre-built models and complex dependencies, or if you are not working with PyTorch.
Stars
84
Forks
11
Language
Python
License
MIT
Category
Last pushed
Mar 13, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/PanJinquan/Pytorch-Base-Trainer"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
deepspeedai/DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference...
helmholtz-analytics/heat
Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
hpcaitech/ColossalAI
Making large AI models cheaper, faster and more accessible
horovod/horovod
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
bsc-wdc/dislib
The Distributed Computing library for python implemented using PyCOMPSs programming model for HPC.